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

x

Browse Source
This commit is contained in:
chenjw28
2026-06-11 17:05:40 +08:00
parent 798ab1df7c
commit 18898057e6
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+3 -3
View File
@@ -295,15 +295,15 @@ RUN set -ex && \
--index-url ${TORCH_INDEX} && \ --index-url ${TORCH_INDEX} && \
${VENV}/bin/python -c "import vllm; print('vllm import OK:', vllm.__version__)" && \ ${VENV}/bin/python -c "import vllm; print('vllm import OK:', vllm.__version__)" && \
# 锁定所有 ROCm 关键包版本，防止 mineru 安装时覆盖 # 锁定所有 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 rm -rf /opt/vllm_build
# =========================================================================== # ===========================================================================
# 阶段 9：安装 MinerU（--no-deps 避免污染 ROCm 环境） # 阶段 9：安装 MinerU（constraint 保护 ROCm 环境，正常安装其他依赖）
# =========================================================================== # ===========================================================================
RUN set -ex && \ RUN set -ex && \
export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \ 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')" ${VENV}/bin/python -c "import torch; import vllm; import mineru; print('MinerU OK')"
# =========================================================================== # ===========================================================================