diff --git a/Dockerfile b/Dockerfile index 54ce09d..6be4d9f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -69,11 +69,11 @@ ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE" ENV ROCM_HOME="/opt/rocm" ENV HIP_PATH="/opt/rocm" ENV PATH="/opt/rocm/bin:/opt/rocm/llvm/bin:$PATH" -ENV LD_LIBRARY_PATH="/opt/rocm/lib:/opt/rocm/lib64:/opt/rocm/llvm/lib:$LD_LIBRARY_PATH" +ENV LD_LIBRARY_PATH="/opt/rocm/lib:/opt/rocm/lib64:/opt/rocm/llvm/lib" RUN git config --global http.proxy socks5://192.168.0.11:1080 && \ git config --global https.proxy socks5://192.168.0.11:1080 && \ - proxychains git clone --recursive https://github.com/ROCm/flash-attention.git && \ + git clone --recursive https://github.com/ROCm/flash-attention.git && \ cd flash-attention && \ git checkout main_perf && \ python setup.py install && \