cjw/mineru-rocm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add proxy for stage 9 PyTorch force-reinstall download

Browse Source
This commit is contained in:
chenjw28
2026-06-09 16:45:59 +08:00
parent be14a07bb7
commit 70c08aa1d5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+1
View File
@@ -295,6 +295,7 @@ RUN set -ex && \
${VENV}/bin/pip install --no-cache-dir 'mineru[core]' && \ ${VENV}/bin/pip install --no-cache-dir 'mineru[core]' && \
${VENV}/bin/python /opt/apply_mineru_patches.py && \ ${VENV}/bin/python /opt/apply_mineru_patches.py && \
# mineru[core] 会把 ROCm PyTorch 替换成 CUDA 版，按教程顺序恢复 # mineru[core] 会把 ROCm PyTorch 替换成 CUDA 版，按教程顺序恢复
export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \
${VENV}/bin/pip install --no-cache-dir --force-reinstall \ ${VENV}/bin/pip install --no-cache-dir --force-reinstall \
torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \ torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \
--index-url ${TORCH_INDEX} && \ --index-url ${TORCH_INDEX} && \