x

2026-03-14 02:44:09 +08:00
parent 7ad5ce520b
commit 1d09d1d4fa
1 changed files with 3 additions and 1 deletions
@@ -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 && \