x

2026-06-05 14:18:13 +08:00
parent 9d0b4b4749
commit 5fc53646b2
1 changed files with 2 additions and 0 deletions
@@ -96,6 +96,8 @@ RUN python${PYTHON_VER} -m venv ${VENV} && \
    echo "[global]" > /root/.pip/pip.conf && \
    echo "index-url = ${PIP_INDEX}" >> /root/.pip/pip.conf && \
    ${VENV}/bin/pip install --no-cache-dir -U pip setuptools wheel && \
+    # PyTorch ROCm wheels 在海外 CDN，通过代理下载
+    export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \
    ${VENV}/bin/pip install --no-cache-dir --pre \
        torch==2.11.0+rocm7.2 \
        torchvision \