cjw/mineru-rocm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
chenjw28
2026-06-11 13:50:59 +08:00
parent b8b11eb353
commit 883c89973c
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+1
View File
@@ -294,6 +294,7 @@ RUN set -ex && \
torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \ torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \
--index-url ${TORCH_INDEX} && \ --index-url ${TORCH_INDEX} && \
${VENV}/bin/python -c "import vllm; print('vllm import OK:', vllm.__version__)" && \ ${VENV}/bin/python -c "import vllm; print('vllm import OK:', vllm.__version__)" && \
${VENV}/bin/pip show vllm | awk '/^Version:/{printf "vllm==%s\n", $2}' >> ${VENV}/torch-constraint.txt && \
rm -rf /opt/vllm_build rm -rf /opt/vllm_build
# =========================================================================== # ===========================================================================