From 4980d9e49af423049011d0f3b707815b3deea5ef Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 3 Apr 2026 08:36:36 +0000 Subject: [PATCH] Use PyTorch nightly with CUDA 13.0 (torch 2.11.0.dev) --- vllm/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/Dockerfile b/vllm/Dockerfile index 12e5a6f..95e57b8 100644 --- a/vllm/Dockerfile +++ b/vllm/Dockerfile @@ -54,8 +54,8 @@ ENV PATH=${CUDA_HOME}/cuda/bin:${PATH} RUN apt-get update && apt install -y wget RUN uv pip install numpy==2.0.0 -# Install PyTorch 2.9.0 with CUDA 13.0 -RUN uv pip install torch==2.9.0+cu130 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu130 +# Install PyTorch nightly with CUDA 13.0 (bleeding edge) +RUN uv pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130 FROM base AS build-base RUN mkdir /wheels