Add CUDA 12.4

ai-dock · Aug 6, 2024 · f352bcc · f352bcc
1 parent f43a324
commit f352bcc
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -92,9 +92,9 @@ jobs:
       fail-fast: false
       matrix:
         build:
-        - {latest: "false", cuda: "11.8.0-base"}
-        - {latest: "false", cuda: "11.8.0-cudnn8-runtime"}
-        - {latest: "false", cuda: "11.8.0-cudnn8-devel"}
+        - {latest: "false", cuda: "12.4.1-base"}
+        - {latest: "false", cuda: "12.4.1-cudnn-runtime"}
+        - {latest: "false", cuda: "12.4.1-cudnn-devel"}
         - {latest: "false", cuda: "12.1.1-base"}
         - {latest: "false", cuda: "12.1.1-cudnn8-runtime"}
         - {latest: "false", cuda: "12.1.1-cudnn8-devel"}
@@ -173,9 +173,6 @@ jobs:
       fail-fast: false
       matrix:
         build:
-        - {latest: "false", rocm: "5.7-core"}
-        - {latest: "false", rocm: "5.7-runtime"}
-        - {latest: "false", rocm: "5.7-devel"}
         - {latest: "false", rocm: "6.0-core"}
         - {latest: "false", rocm: "6.0-runtime"}
         - {latest: "false", rocm: "6.0-devel"}