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 14:49:05 +08:00
parent 83159a6de9
commit 827a1aa977
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+3 -1
View File
@@ -309,7 +309,9 @@ RUN set -ex && \
${VENV}/bin/pip install --force-reinstall \ ${VENV}/bin/pip install --force-reinstall \
torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \ torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \
--index-url ${TORCH_INDEX} && \ --index-url ${TORCH_INDEX} && \
cd /opt/vllm && ${VENV}/bin/pip install -e . --no-build-isolation --no-deps && \ cd /opt/vllm && \
sed -i 's/license = "Apache-2.0"/license = {text = "Apache-2.0"}/' pyproject.toml && \
${VENV}/bin/pip install -e . --no-build-isolation --no-deps && \
${VENV}/bin/python -c "import triton.language; print('triton OK')" && \ ${VENV}/bin/python -c "import triton.language; print('triton OK')" && \
${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)" && \ ${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)" && \
${VENV}/bin/python -c "import vllm; print('vllm OK:', vllm.__version__)" ${VENV}/bin/python -c "import vllm; print('vllm OK:', vllm.__version__)"