Skip to content

Commit 767eee6

Browse files
authored
Pinned to pytorch 2.8 (sgl-project#10)
1 parent b3311b4 commit 767eee6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Dockerfile.xpu_kernel

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ RUN --mount=type=secret,id=github_token \
3838
conda activate py${PYTHON_VERSION} && \
3939
# . /opt/intel/oneapi/setvars.sh --force && \
4040
# Install Torch
41-
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
41+
pip install torch==2.8.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/xpu
4242

4343
# Install vllm from source
4444
RUN --mount=type=secret,id=github_token \

0 commit comments

Comments
 (0)