From c69349052a933365a19b04730c1833639a009e29 Mon Sep 17 00:00:00 2001 From: chenjw28 <792430652@qq.com> Date: Wed, 10 Jun 2026 17:54:52 +0800 Subject: [PATCH] x --- docker/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index ab41021..7822956 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -285,11 +285,11 @@ COPY scripts/cache_warmer.py /opt/scripts/cache_warmer.py # 阶段 8b:安装 vllm + 平台补丁 + 验证 # =========================================================================== RUN set -ex && \ + export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \ cd /opt/vllm && ${VENV}/bin/pip install -e . --no-build-isolation && \ ${VENV}/bin/pip install regex && \ ${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)" && \ ${VENV}/bin/pip uninstall -y triton triton-rocm 2>/dev/null; \ - export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \ ${VENV}/bin/pip install --force-reinstall \ torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \ --index-url ${TORCH_INDEX} && \ @@ -300,11 +300,11 @@ RUN set -ex && \ # 阶段 9:安装 MinerU + RDNA 适配补丁 + PyTorch 恢复 # =========================================================================== 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 版 ${VENV}/bin/pip uninstall -y triton triton-rocm 2>/dev/null; \ - export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \ ${VENV}/bin/pip install --force-reinstall \ torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \ --index-url ${TORCH_INDEX} && \