feat(cache): update cuda & cudnn version (#1943)

Signed-off-by: Keming <[email protected]>
tensorchord · Jan 9, 2025 · 95bb1c3 · 95bb1c3
1 parent 6d654a2
commit 95bb1c3
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -174,16 +174,15 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        build_func: ["build", "build_gpu_11_2", "build_gpu_11_3", "build_gpu_11_6"]
         include:
           - build_func: build
             tag_suffix: ""
-          - build_func: build_gpu_11_2
-            tag_suffix: "-cuda-11.2.2-cudnn-8"
-          - build_func: build_gpu_11_3
-            tag_suffix: "-cuda-11.3.1-cudnn-8"
           - build_func: build_gpu_11_6
             tag_suffix: "-cuda-11.6.2-cudnn-8"
+          - build_func: build_gpu_11_8
+            tag_suffix: "-cuda-11.8.0-cudnn-8"
+          - build_func: build_gpu_12_3
+            tag_suffix: "-cuda-12.3.2-cudnn-9"
     needs: image_publish
     steps:
     - uses: actions/checkout@v4
@@ -216,5 +215,5 @@ jobs:
       run: ./base-images/remote-cache/build-and-push-remote-cache.sh
       env:
         BUILD_FUNC: ${{ matrix.build_func }}
-        TAG_SUFFIX: ${{ matrix.TAG_SUFFIX }}
+        TAG_SUFFIX: ${{ matrix.tag_suffix }}
         ENVD_BUILD_OWNER: 1000
diff --git a/base-images/remote-cache/build.envd b/base-images/remote-cache/build.envd
@@ -18,3 +18,15 @@ def build_gpu_11_6():
     """pytorch"""
     base(os="ubuntu20.04", language="python3")
     install.cuda(version="11.6.2", cudnn="8")
+
+
+def build_gpu_11_8():
+    """pytorch"""
+    base(os="ubuntu20.04", language="python3")
+    install.cuda(version="11.8.0", cudnn="8")
+
+
+def build_gpu_12_3():
+    """pytorch"""
+    base(os="ubuntu20.04", language="python3")
+    install.cuda(version="12.3.2", cudnn="9")