From 05e40a6876c76b3f008f5b2304f5eabb529f3efb Mon Sep 17 00:00:00 2001
From: Romain Avouac <43444134+avouacr@users.noreply.github.com>
Date: Sat, 25 Nov 2023 21:42:30 +0100
Subject: [PATCH] Cuda 11.8 -> Cuda 12.1 (#160)

* build for cuda 12

* cuda 12.1.1

* Update base.json
---
 .github/workflows/main-workflow.yml         | 2 +-
 .github/workflows/pull-request-workflow.yml | 2 +-
 python-pytorch/Dockerfile                   | 2 +-
 releases/base.json                          | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/main-workflow.yml b/.github/workflows/main-workflow.yml
index e176a593..1543ae2a 100644
--- a/.github/workflows/main-workflow.yml
+++ b/.github/workflows/main-workflow.yml
@@ -14,7 +14,7 @@ jobs:
       context: base
       test: true
       base_image: ubuntu:22.04
-      base_image_gpu: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+      base_image_gpu: nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
       external_base_image: true
       branch: ${{ github.ref }}
     secrets:
diff --git a/.github/workflows/pull-request-workflow.yml b/.github/workflows/pull-request-workflow.yml
index 017976d2..ec0f5f82 100644
--- a/.github/workflows/pull-request-workflow.yml
+++ b/.github/workflows/pull-request-workflow.yml
@@ -63,7 +63,7 @@ jobs:
       context: base
       test: true
       base_image: ubuntu:22.04
-      base_image_gpu: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+      base_image_gpu: nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
       external_base_image: true
       branch: ${{ github.ref }}
   python-minimal:
diff --git a/python-pytorch/Dockerfile b/python-pytorch/Dockerfile
index d97ac2ef..b2596a4d 100644
--- a/python-pytorch/Dockerfile
+++ b/python-pytorch/Dockerfile
@@ -7,7 +7,7 @@ USER root
 
 RUN if ! [[ -z "${CUDA_VERSION}" ]]; then \
         # CUDA support
-        pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118; \
+        pip3 install torch torchvision torchaudio; \
     else \
         # CPU only
         pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu; \
diff --git a/releases/base.json b/releases/base.json
index 79e41c15..c363a8e0 100644
--- a/releases/base.json
+++ b/releases/base.json
@@ -4,7 +4,7 @@
         "output_image":"base:latest"
     },
     {
-        "base_image":"nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04",
+        "base_image":"nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04",
         "output_image":"base:latest",
         "gpu":"true"
     }