From 926dd57a871ebc415d43dfca5637b52c4ed98183 Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Sat, 21 Mar 2026 15:15:01 -0700 Subject: [PATCH] cuda 13.2 torch --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index ef0d766..c517952 100644 --- a/Dockerfile +++ b/Dockerfile @@ -45,7 +45,7 @@ RUN apt update && \ # Additional deps RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ - uv pip install torch torchvision torchaudio triton --index-url https://download.pytorch.org/whl/nightly/cu130 && \ + uv pip install torch torchvision torchaudio triton --index-url https://download.pytorch.org/whl/nightly/cu132 && \ uv pip install nvidia-nvshmem-cu13 "apache-tvm-ffi<0.2" filelock pynvml requests tqdm # Configure Ccache for CUDA/C++ @@ -256,7 +256,7 @@ ARG PRE_TRANSFORMERS=0 # Install deps RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ - uv pip install torch torchvision torchaudio triton --index-url https://download.pytorch.org/whl/nightly/cu130 && \ + uv pip install torch torchvision torchaudio triton --index-url https://download.pytorch.org/whl/nightly/cu132 && \ uv pip install nvidia-nvshmem-cu13 "apache-tvm-ffi<0.2" # Install wheels from host ./wheels/ (bind-mounted from build context — no layer bloat)