Skip to content

Commit 91c1fe1

Browse files
authored
Merge pull request #921 from docker/vllm-cuda13-default
chore: update vllm install to use --torch-backend auto
2 parents 1768937 + 9f305f8 commit 91c1fe1

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

Dockerfile

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -102,10 +102,7 @@ USER modelrunner
102102
RUN curl -LsSf https://astral.sh/uv/install.sh | sh \
103103
&& ~/.local/bin/uv venv --python 3.12 /opt/vllm-env \
104104
&& . /opt/vllm-env/bin/activate \
105-
&& ~/.local/bin/uv pip install vllm \
106-
--extra-index-url https://wheels.vllm.ai/${VLLM_VERSION}/cu130 \
107-
--extra-index-url https://download.pytorch.org/whl/cu130 \
108-
--index-strategy unsafe-best-match
105+
&& ~/.local/bin/uv pip install vllm --torch-backend auto
109106

110107
RUN /opt/vllm-env/bin/python3.12 -c "import vllm; print(vllm.__version__)" > /opt/vllm-env/version
111108

0 commit comments

Comments
 (0)