diff --git a/docker/Dockerfile b/docker/Dockerfile
index 7822956..1fdd977 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -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
 
 # ===========================================================================