From 5b8d531acae5ef4f56a75961d68bfdaaf1ea0fb8 Mon Sep 17 00:00:00 2001 From: Thijs Walcarius Date: Sat, 26 Apr 2025 02:34:29 +0200 Subject: [PATCH] Upgrade cuda12 version to 12.8 (#2288) Co-authored-by: Ayaz Salikhov --- 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 d82b5b51..40231441 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/cu126' \ +RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu128' \ 'torch' \ 'torchaudio' \ 'torchvision' && \