diff --git a/docker/Dockerfile b/docker/Dockerfile index a4d5f22..608ccde 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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 \