Use PyTorch nightly with CUDA 13.0 (torch 2.11.0.dev)

This commit is contained in:
2026-04-03 08:36:36 +00:00
parent 6a97539682
commit 4980d9e49a

View File

@@ -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