Cuda 11.8 -> Cuda 12.1 (InseeFrLab#160)

* build for cuda 12 * cuda 12.1.1 * Update base.json
phlg · Nov 25, 2023 · 05e40a6 · 05e40a6
1 parent 5307c8e
commit 05e40a6
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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"
     }