fix: cuda tag in image cache (#1156)

Signed-off-by: Keming <kemingyang@tensorchord.ai> Signed-off-by: Keming <kemingyang@tensorchord.ai>
tensorchord · Nov 4, 2022 · 6b7a0e9 · 6b7a0e9
1 parent 1af3449
commit 6b7a0e9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -154,11 +154,11 @@ jobs:
           - build_func: build
             tag_suffix: ""
           - build_func: build_gpu_11_2
-            tag_suffix: "-cuda-11.2.0-cudnn-8"
+            tag_suffix: "-cuda-11.2.2-cudnn-8"
           - build_func: build_gpu_11_3
-            tag_suffix: "-cuda-11.3.0-cudnn-8"
+            tag_suffix: "-cuda-11.3.1-cudnn-8"
           - build_func: build_gpu_11_6
-            tag_suffix: "-cuda-11.6.0-cudnn-8"
+            tag_suffix: "-cuda-11.6.2-cudnn-8"
     needs: image_publish
     steps:
     - uses: actions/checkout@v3

diff --git a/base-images/remote-cache/build-and-push-remote-cache.sh b/base-images/remote-cache/build-and-push-remote-cache.sh
@@ -27,8 +27,5 @@ TAG_SUFFIX="${TAG_SUFFIX:-}"
 cd ${ROOT_DIR}
 
 envd --debug build -f build.envd:${BUILD_FUNC} --export-cache type=registry,ref=docker.io/${DOCKER_HUB_ORG}/python-cache:envd-v${ENVD_VERSION}${TAG_SUFFIX} --force
-# envd build -f build.envd:build_gpu_11_2 --export-cache type=registry,ref=docker.io/${DOCKER_HUB_ORG}/python-cache:envd-v${ENVD_VERSION}-cuda-11.2.0-cudnn-8 --force
-# envd build -f build.envd:build_gpu_11_3 --export-cache type=registry,ref=docker.io/${DOCKER_HUB_ORG}/python-cache:envd-v${ENVD_VERSION}-cuda-11.3.0-cudnn-8 --force
-# envd build -f build.envd:build_gpu_11_6 --export-cache type=registry,ref=docker.io/${DOCKER_HUB_ORG}/python-cache:envd-v${ENVD_VERSION}-cuda-11.6.0-cudnn-8 --force
 
 cd - > /dev/null