From 99bd3f8a2484f8e409be622e0017466c16cc1c73 Mon Sep 17 00:00:00 2001 From: Ayaz Salikhov Date: Fri, 25 Apr 2025 01:39:58 +0100 Subject: [PATCH] Upgrade CUDA version used in PyTorch cuda12-variant to 12.6 (#2291) --- images/pytorch-notebook/cuda12/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/images/pytorch-notebook/cuda12/Dockerfile b/images/pytorch-notebook/cuda12/Dockerfile index 983c439a..d82b5b51 100644 --- a/images/pytorch-notebook/cuda12/Dockerfile +++ b/images/pytorch-notebook/cuda12/Dockerfile @@ -13,7 +13,7 @@ SHELL ["/bin/bash", "-o", "pipefail", "-c"] # Install PyTorch with pip (https://pytorch.org/get-started/locally/) # hadolint ignore=DL3013 -RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu124' \ +RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu126' \ 'torch' \ 'torchaudio' \ 'torchvision' && \