Use PyTorch nightly with CUDA 13.0 (torch 2.11.0.dev)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user