From 9221620c20766349f8487cd38ebe72253314e585 Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Tue, 1 Oct 2024 10:10:35 -0400 Subject: [PATCH] [Docs] Support CUDA 12.3 packages; deprecate 12.1 packages (#2957) This PR updates documents to add the CUDA 12.3 packages and deprecate the 12.1 package support. We need CUDA 12.3 for some CUTLASS kernels in H100 GPUs. --- docs/install/mlc_llm.rst | 8 ++++---- docs/install/tvm.rst | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/docs/install/mlc_llm.rst b/docs/install/mlc_llm.rst index 2688e836d9..415bd6682d 100644 --- a/docs/install/mlc_llm.rst +++ b/docs/install/mlc_llm.rst @@ -32,19 +32,19 @@ Select your operating system/compute platform and run the command in your termin conda activate your-environment python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cpu mlc-ai-nightly-cpu - .. tab:: CUDA 12.1 + .. tab:: CUDA 12.2 .. code-block:: bash conda activate your-environment - python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu121 mlc-ai-nightly-cu121 + python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu122 mlc-ai-nightly-cu122 - .. tab:: CUDA 12.2 + .. tab:: CUDA 12.3 .. code-block:: bash conda activate your-environment - python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu122 mlc-ai-nightly-cu122 + python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cu123 mlc-ai-nightly-cu123 .. tab:: ROCm 6.1 diff --git a/docs/install/tvm.rst b/docs/install/tvm.rst index bd802b8d11..901184e233 100644 --- a/docs/install/tvm.rst +++ b/docs/install/tvm.rst @@ -39,19 +39,19 @@ A nightly prebuilt Python package of Apache TVM Unity is provided. conda activate your-environment python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu - .. tab:: CUDA 12.1 + .. tab:: CUDA 12.2 .. code-block:: bash conda activate your-environment - python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121 + python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu122 - .. tab:: CUDA 12.2 + .. tab:: CUDA 12.3 .. code-block:: bash conda activate your-environment - python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu122 + python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123 .. tab:: ROCm 6.1