build for cuda 12

InseeFrLab · Nov 24, 2023 · 44ef3f0 · 44ef3f0
1 parent 5307c8e
commit 44ef3f0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 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.0-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.0-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; \