From 83159a6de9c26a191707174af4a621e0b242b366 Mon Sep 17 00:00:00 2001
From: chenjw28 <792430652@qq.com>
Date: Thu, 11 Jun 2026 14:32:41 +0800
Subject: [PATCH] x

---
 docker/Dockerfile | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 1fdd977..b264d58 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -303,14 +303,16 @@ RUN set -ex && \
 RUN set -ex && \
     export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \
     ${VENV}/bin/pip install 'mineru[core]' && \
-    # mineru[core] 会把 ROCm PyTorch 替换成 CUDA 版 + 安装 CUDA triton
-    # 必须先卸载 CUDA triton，再装回 ROCm 版
+    # mineru[core] 会把 ROCm PyTorch/vllm 替换成 CUDA 版
+    # 必须先卸载 CUDA triton，再装回 ROCm 版 torch + vllm
     ${VENV}/bin/pip uninstall -y triton triton-rocm 2>/dev/null; \
     ${VENV}/bin/pip install --force-reinstall \
         torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \
         --index-url ${TORCH_INDEX} && \
+    cd /opt/vllm && ${VENV}/bin/pip install -e . --no-build-isolation --no-deps && \
     ${VENV}/bin/python -c "import triton.language; print('triton OK')" && \
-    ${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)"
+    ${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)" && \
+    ${VENV}/bin/python -c "import vllm; print('vllm OK:', vllm.__version__)"
 
 # ===========================================================================
 # 阶段 10：入口与最终验证