software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Reverted to Torch 2.9.1 in the source build to address #24

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-01-30 10:43:12 -08:00
parent 34bd3ae39c
commit 7d232a305a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@@ -70,7 +70,7 @@ ARG CACHEBUST_DEPS=1
# pip reuses previously downloaded wheels. # pip reuses previously downloaded wheels.
RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 uv pip install torch==2.9.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130
# Install additional dependencies # Install additional dependencies
RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \