x

2026-06-11 13:50:59 +08:00
parent b8b11eb353
commit 883c89973c
1 changed files with 1 additions and 0 deletions
@@ -294,6 +294,7 @@ RUN set -ex && \
        torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \
        --index-url ${TORCH_INDEX} && \
    ${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

 # ===========================================================================