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:
chenjw28
2026-03-11 09:49:10 +08:00
parent 51c53811a0
commit 65c768dfb2
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+2 -5
View File
@@ -64,11 +64,8 @@ RUN python -m pip install \
WORKDIR /opt WORKDIR /opt
ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE" ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
RUN git clone https://github.com/ROCm/flash-attention.git &&\ RUN pip install flash-attn --no-build-isolation \
cd flash-attention &&\ --index-url https://pypi.tuna.tsinghua.edu.cn/simple
git checkout main_perf &&\
python setup.py install && \
cd /opt && rm -rf /opt/flash-attention
# 6. Clone vLLM # 6. Clone vLLM
RUN git clone https://github.com/vllm-project/vllm.git /opt/vllm RUN git clone https://github.com/vllm-project/vllm.git /opt/vllm