Skip to content

Commit 558b764

Browse files
authored
[Doc] Update pip install commands for CUDA (#3343)
This PR updates the pip install instructions for CUDA in documentation, where we prioritize CUDA 12.8 and 13.0.
1 parent 4439e05 commit 558b764

File tree

3 files changed

+11
-11
lines changed

3 files changed

+11
-11
lines changed

ci/task/test_model_compile.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ pip install --force-reinstall wheels/*.whl
99

1010
if [[ ${GPU} == cuda* ]]; then
1111
TARGET=cuda
12-
pip install --pre -U --no-index -f https://mlc.ai/wheels mlc-ai-nightly-cu123
12+
pip install --pre -U --no-index -f https://mlc.ai/wheels mlc-ai-nightly-cu128
1313
export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
1414
elif [[ ${GPU} == rocm* ]]; then
1515
TARGET=rocm

docs/install/mlc_llm.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,19 +32,19 @@ Select your operating system/compute platform and run the command in your termin
3232
conda activate your-environment
3333
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cpu mlc-ai-nightly-cpu
3434
35-
.. tab:: CUDA 12.2
35+
.. tab:: CUDA 12.8
3636

3737
.. code-block:: bash
3838
3939
conda activate your-environment
40-
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu122 mlc-ai-nightly-cu122
40+
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu128 mlc-ai-nightly-cu128
4141
42-
.. tab:: CUDA 12.3
42+
.. tab:: CUDA 13.0
4343

4444
.. code-block:: bash
4545
4646
conda activate your-environment
47-
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu123 mlc-ai-nightly-cu123
47+
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu130 mlc-ai-nightly-cu130
4848
4949
.. tab:: ROCm 6.1
5050

@@ -80,7 +80,7 @@ Select your operating system/compute platform and run the command in your termin
8080

8181
.. code-block:: bash
8282
83-
conda install -c conda-forge libgcc-ng
83+
conda install -c conda-forge libstdcxx-ng
8484
8585
Besides, we would recommend using Python 3.11; so if you are creating a new environment,
8686
you could use the following command:

docs/install/tvm.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,19 +39,19 @@ A nightly prebuilt Python package of Apache TVM Unity is provided.
3939
conda activate your-environment
4040
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
4141
42-
.. tab:: CUDA 12.2
42+
.. tab:: CUDA 12.8
4343

4444
.. code-block:: bash
4545
4646
conda activate your-environment
47-
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu122
47+
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu128
4848
49-
.. tab:: CUDA 12.3
49+
.. tab:: CUDA 13.0
5050

5151
.. code-block:: bash
5252
5353
conda activate your-environment
54-
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
54+
python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu130
5555
5656
.. tab:: ROCm 6.1
5757

@@ -77,7 +77,7 @@ A nightly prebuilt Python package of Apache TVM Unity is provided.
7777

7878
.. code-block:: bash
7979
80-
conda install -c conda-forge libgcc-ng
80+
conda install -c conda-forge libstdcxx-ng
8181
8282
.. tab:: macOS
8383

0 commit comments

Comments
 (0)