[CI] Update docker image (#2956)

This PR updates the CI docker image to deprecate the CUDA 12.1 in CI.
mlc-ai · Oct 1, 2024 · 5dee045 · 5dee045
1 parent d516292
commit 5dee045
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ci/jenkinsfile.groovy b/ci/jenkinsfile.groovy
@@ -21,9 +21,9 @@ run_cpu = "bash ci/bash.sh mlcaidev/ci-cpu:4d61e5d -e GPU cpu -e MLC_CI_SETUP_DE
 run_cuda = "bash ci/bash.sh mlcaidev/ci-cu121:4d61e5d -e GPU cuda-12.1 -e MLC_CI_SETUP_DEPS 1"
 run_rocm = "bash ci/bash.sh mlcaidev/ci-rocm57:4d61e5d -e GPU rocm-5.7 -e MLC_CI_SETUP_DEPS 1"
 
-pkg_cpu = "bash ci/bash.sh mlcaidev/package-rocm61:b6b04f1 -e GPU cpu -e MLC_CI_SETUP_DEPS 1"
-pkg_cuda = "bash ci/bash.sh mlcaidev/package-cu121:b6b04f1 -e GPU cuda-12.1 -e MLC_CI_SETUP_DEPS 1"
-pkg_rocm = "bash ci/bash.sh mlcaidev/package-rocm61:b6b04f1 -e GPU rocm-6.1 -e MLC_CI_SETUP_DEPS 1"
+pkg_cpu = "bash ci/bash.sh mlcaidev/package-rocm61:254d630 -e GPU cpu -e MLC_CI_SETUP_DEPS 1"
+pkg_cuda = "bash ci/bash.sh mlcaidev/package-cu122:254d630 -e GPU cuda-12.2 -e MLC_CI_SETUP_DEPS 1"
+pkg_rocm = "bash ci/bash.sh mlcaidev/package-rocm61:254d630 -e GPU rocm-6.1 -e MLC_CI_SETUP_DEPS 1"
 
 
 def per_exec_ws(folder) {