cjw/vllm-r9700-container
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
SZCJW
2026-03-14 02:51:49 +08:00
parent 1d09d1d4fa
commit 2c2617e219
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+2 -2
View File
@@ -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 && \