From be14a07bb7c74a45f7408fc83b5f13cab2ea848e Mon Sep 17 00:00:00 2001
From: chenjw28 <792430652@qq.com>
Date: Tue, 9 Jun 2026 16:35:10 +0800
Subject: [PATCH] fix: add PyTorch recovery after mineru[core] install (CUDA
 overwrite)

---
 docker/Dockerfile | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 9d3a6f6..3670bb2 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -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：入口与最终验证