[CI] Update TVM package for CI (#3067)

This PR updates the TVM package used for CI from cuda 12.1 to 12.3.
mlc-ai · Dec 16, 2024 · 49dcd4a · 49dcd4a
1 parent 9dd0788
commit 49dcd4a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ci/task/test_model_compile.sh b/ci/task/test_model_compile.sh
@@ -9,7 +9,7 @@ pip install --force-reinstall wheels/*.whl
 
 if [[ ${GPU} == cuda* ]]; then
 	TARGET=cuda
-	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121
+	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
 	export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
 elif [[ ${GPU} == rocm* ]]; then
 	TARGET=rocm

diff --git a/ci/task/test_unittest.sh b/ci/task/test_unittest.sh
@@ -8,7 +8,7 @@ if [[ -n ${MLC_CI_SETUP_DEPS:-} ]]; then
     # Install dependency
     pip install --force-reinstall wheels/*.whl
     pip install --quiet pytest
-    pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121
+    pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
     export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
 fi