x

2026-03-11 09:49:10 +08:00
parent 51c53811a0
commit 65c768dfb2
1 changed files with 2 additions and 5 deletions
@@ -64,11 +64,8 @@ RUN python -m pip install \
 WORKDIR /opt
 ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"

-RUN git clone https://github.com/ROCm/flash-attention.git &&\
-  cd flash-attention &&\
-  git checkout main_perf &&\
-  python setup.py install && \
-  cd /opt && rm -rf /opt/flash-attention
+RUN pip install flash-attn --no-build-isolation \
+  --index-url https://pypi.tuna.tsinghua.edu.cn/simple

 # 6. Clone vLLM
 RUN git clone https://github.com/vllm-project/vllm.git /opt/vllm