diff --git a/Dockerfile b/Dockerfile index 59f25c3..54ce09d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -71,7 +71,9 @@ 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" -RUN proxychains git clone https://github.com/ROCm/flash-attention.git && \ +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 && \ cd flash-attention && \ git checkout main_perf && \ python setup.py install && \