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-10 17:18:52 +08:00
parent a94374fbb1
commit f5a9b786d0
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+4
View File
@@ -97,6 +97,8 @@ RUN python${PYTHON_VER} -m venv ${VENV} && \
mkdir -p /root/.pip && \ mkdir -p /root/.pip && \
echo "[global]" > /root/.pip/pip.conf && \ echo "[global]" > /root/.pip/pip.conf && \
echo "index-url = ${PIP_INDEX}" >> /root/.pip/pip.conf && \ echo "index-url = ${PIP_INDEX}" >> /root/.pip/pip.conf && \
echo "[install]" >> /root/.pip/pip.conf && \
echo "constraint = ${VENV}/torch-constraint.txt" >> /root/.pip/pip.conf && \
${VENV}/bin/pip install -U pip setuptools wheel && \ ${VENV}/bin/pip install -U pip setuptools wheel && \
# PyTorch ROCm wheels 在海外 CDN，通过代理下载 # PyTorch ROCm wheels 在海外 CDN，通过代理下载
export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \ export http_proxy=${GIT_PROXY} https_proxy=${GIT_PROXY} && \
@@ -105,6 +107,8 @@ RUN python${PYTHON_VER} -m venv ${VENV} && \
torchvision \ torchvision \
pytorch-triton-rocm \ pytorch-triton-rocm \
--index-url ${TORCH_INDEX} && \ --index-url ${TORCH_INDEX} && \
# 锁定 torch 版本，防止后续 pip install 解析到 CUDA 版
echo "torch==2.11.0+rocm7.2" > ${VENV}/torch-constraint.txt && \
${VENV}/bin/python -c "import torch; print('PyTorch:', torch.__version__); assert torch.version.hip is not None" ${VENV}/bin/python -c "import torch; print('PyTorch:', torch.__version__); assert torch.version.hip is not None"
# =========================================================================== # ===========================================================================