refactor: build cuda12 instead of cuda tag

Author: zynaa

.github/workflows/docker.yml

@@ -232,23 +232,23 @@ jobs:
     needs: [x86_64-scipy]
     if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
 
-  aarch64-pytorch-cuda:
+  aarch64-pytorch-cuda12:
     uses: ./.github/workflows/docker-build-test-upload.yml
     with:
       parent-image: scipy-notebook
       image: pytorch-notebook
-      variant: cuda
+      variant: cuda12
       platform: aarch64
       runs-on: ARM64_SLOW
     needs: [aarch64-scipy]
     if: github.repository_owner == 'jupyter' && !contains(github.event.pull_request.title, '[FAST_BUILD]')
 
-  x86_64-pytorch-cuda:
+  x86_64-pytorch-cuda12:
     uses: ./.github/workflows/docker-build-test-upload.yml
     with:
       parent-image: scipy-notebook
       image: pytorch-notebook
-      variant: cuda
+      variant: cuda12
       platform: x86_64
       runs-on: ubuntu-latest
     needs: [x86_64-scipy]
@@ -363,7 +363,7 @@ jobs:
             { image: julia-notebook, variant: default },
             { image: tensorflow-notebook, variant: default },
             { image: pytorch-notebook, variant: default },
-            { image: pytorch-notebook, variant: cuda },
+            { image: pytorch-notebook, variant: cuda12 },
             { image: pytorch-notebook, variant: cuda11 },
             { image: datascience-notebook, variant: default },
             { image: pyspark-notebook, variant: default },
@@ -425,7 +425,7 @@ jobs:
             { image: julia-notebook, variant: default },
             { image: tensorflow-notebook, variant: default },
             { image: pytorch-notebook, variant: default },
-            { image: pytorch-notebook, variant: cuda },
+            { image: pytorch-notebook, variant: cuda12 },
             { image: pytorch-notebook, variant: cuda11 },
             { image: datascience-notebook, variant: default },
             { image: pyspark-notebook, variant: default },
@@ -486,7 +486,7 @@ jobs:
             { image: julia-notebook, variant: default },
             { image: tensorflow-notebook, variant: default },
             { image: pytorch-notebook, variant: default },
-            { image: pytorch-notebook, variant: cuda },
+            { image: pytorch-notebook, variant: cuda12 },
             { image: pytorch-notebook, variant: cuda11 },
             { image: datascience-notebook, variant: default },
             { image: pyspark-notebook, variant: default },

images/pytorch-notebook/cuda/Dockerfile renamed to 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 \
+RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu121' \
     'torch' \
     'torchvision' \
     'torchaudio'  && \