x

docker/Dockerfile

@@ -20,6 +20,7 @@ ARG PIP_INDEX=https://mirrors.aliyun.com/pypi/simple
 # -- 环境变量 ---------------------------------------------------------------
 ENV DEBIAN_FRONTEND=noninteractive \
     PATH=/opt/rocm/bin:/opt/rocm/llvm/bin:${VENV}/bin:${PATH} \
+    PYTHONPATH=/opt/vllm:${PYTHONPATH} \
     PYTORCH_ROCM_ARCH=${ARCH} \
     FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE \
     MINERU_MODEL_SOURCE=huggingface \
@@ -304,7 +305,6 @@ RUN set -ex && \
 RUN set -ex && \
     export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \
     ${VENV}/bin/pip install 'mineru[core]' click gradio && \
-    cd /opt/vllm && ${VENV}/bin/pip install -e . --no-build-isolation --no-deps && \
     ${VENV}/bin/python -c "import torch; import vllm; import mineru; import click; import gradio; import mineru.cli.gradio_app; print('MinerU + Gradio CLI imports OK')" && \
     ${VENV}/bin/mineru-gradio --help >/dev/null