File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -9,7 +9,7 @@ pip install --force-reinstall wheels/*.whl
9
9
10
10
if [[ ${GPU} == cuda* ]]; then
11
11
TARGET=cuda
12
- pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121
12
+ pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
13
13
export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
14
14
elif [[ ${GPU} == rocm* ]]; then
15
15
TARGET=rocm
Original file line number Diff line number Diff line change @@ -8,7 +8,7 @@ if [[ -n ${MLC_CI_SETUP_DEPS:-} ]]; then
8
8
# Install dependency
9
9
pip install --force-reinstall wheels/* .whl
10
10
pip install --quiet pytest
11
- pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121
11
+ pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
12
12
export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
13
13
fi
14
14
You can’t perform that action at this time.
0 commit comments