x

2026-06-16 11:46:53 +08:00
parent 112dfee95c
commit 7a8bf3b5f9
2 changed files with 68 additions and 0 deletions
@@ -294,6 +294,8 @@ 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__)" && \
+    # 确保 mineru-api 子进程不依赖 PYTHONPATH 也能导入 vllm
+    echo "/opt/vllm" > ${VENV}/lib/python${PYTHON_VER}/site-packages/vllm.pth && \
    # 锁定所有 ROCm 关键包版本，防止 mineru 安装时覆盖
    ${VENV}/bin/pip freeze | grep -iE "^(torch|vllm|triton|pytorch.triton|setuptools|torchvision|flash.attn|aiter)" >> ${VENV}/torch-constraint.txt && \
    rm -rf /opt/vllm_build