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