From 18898057e68cfff9b3a43d4dda32dec52a0b3c12 Mon Sep 17 00:00:00 2001
From: chenjw28 <792430652@qq.com>
Date: Thu, 11 Jun 2026 17:05:40 +0800
Subject: [PATCH] x

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

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 994f112..6b8f58b 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -295,15 +295,15 @@ RUN set -ex && \
         --index-url ${TORCH_INDEX} && \
     ${VENV}/bin/python -c "import vllm; print('vllm import OK:', vllm.__version__)" && \
     # 锁定所有 ROCm 关键包版本，防止 mineru 安装时覆盖
-    ${VENV}/bin/pip freeze | grep -iE "^(torch|vllm|triton|pytorch.triton|setuptools)" >> ${VENV}/torch-constraint.txt && \
+    ${VENV}/bin/pip freeze | grep -iE "^(torch|vllm|triton|pytorch.triton|setuptools|torchvision|flash.attn|aiter)" >> ${VENV}/torch-constraint.txt && \
     rm -rf /opt/vllm_build
 
 # ===========================================================================
-# 阶段 9：安装 MinerU（--no-deps 避免污染 ROCm 环境）
+# 阶段 9：安装 MinerU（constraint 保护 ROCm 环境，正常安装其他依赖）
 # ===========================================================================
 RUN set -ex && \
     export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \
-    ${VENV}/bin/pip install --no-deps 'mineru[core]' && \
+    ${VENV}/bin/pip install 'mineru[core]' && \
     ${VENV}/bin/python -c "import torch; import vllm; import mineru; print('MinerU OK')"
 
 # ===========================================================================