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

fix: add PyTorch recovery after mineru[core] install (CUDA overwrite)

Browse Source
This commit is contained in:
chenjw28
2026-06-09 16:35:10 +08:00
parent 8d3e02a10d
commit be14a07bb7
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+8 -3
View File
@@ -289,12 +289,17 @@ RUN set -ex && \
rm -rf /opt/vllm_build
# ===========================================================================
# 阶段 9：安装 MinerU + RDNA 适配补丁
# 阶段 9：安装 MinerU + RDNA 适配补丁 + PyTorch 恢复
# ===========================================================================
RUN set -ex && \
${VENV}/bin/pip install --no-cache-dir 'mineru[core]' && \
${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)" && \
${VENV}/bin/python /opt/apply_mineru_patches.py
${VENV}/bin/python /opt/apply_mineru_patches.py && \
# mineru[core] 会把 ROCm PyTorch 替换成 CUDA 版，按教程顺序恢复
${VENV}/bin/pip install --no-cache-dir --force-reinstall \
torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \
--index-url ${TORCH_INDEX} && \
${VENV}/bin/pip uninstall -y triton triton-rocm 2>/dev/null; \
${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)"
# ===========================================================================
# 阶段 10：入口与最终验证