x
This commit is contained in:
+3
-1
@@ -309,7 +309,9 @@ RUN set -ex && \
|
||||
${VENV}/bin/pip install --force-reinstall \
|
||||
torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \
|
||||
--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 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__)"
|
||||
|
||||
Reference in New Issue
Block a user