b1d6e2c9b3
* v4.3 update. * Update the cute_dsl_api changelog's doc link * Update version to 4.3.0 * Update the example link * Update doc to encourage user to install DSL from requirements.txt --------- Co-authored-by: Larry Wu <larwu@nvidia.com>
46 lines
1.5 KiB
ReStructuredText
46 lines
1.5 KiB
ReStructuredText
.. _quick_start:
|
|
|
|
Quick Start Guide
|
|
=======================
|
|
|
|
The CUTLASS DSL 4.0 release currently supports **Linux** and **Python 3.12** only. To install CUTLASS DSLs (limited to CuTe DSL for now), use the following command
|
|
|
|
Installation
|
|
-----------------------
|
|
|
|
To ensure compatibility with the examples and code on `GitHub <https://github.com/NVIDIA/cutlass/tree/main>`_,
|
|
use the `requirements.txt <https://github.com/NVIDIA/cutlass/blob/main/python/CuTeDSL/requirements.txt>`_ file from the corresponding commit in the repository.
|
|
|
|
.. code-block:: bash
|
|
|
|
git clone https://github.com/NVIDIA/cutlass.git
|
|
pip install -r cutlass/python/CuTeDSL/requirements.txt
|
|
|
|
If you just want to try out the last known stable release of the CUTLASS DSL (may not compatible with the latest examples and code), run:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install nvidia-cutlass-dsl
|
|
|
|
The ``nvidia-cutlass-dsl`` wheel includes everything needed to generate GPU kernels. It requires
|
|
the same NVIDIA driver version as the
|
|
`CUDA Toolkit 12.9 <https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html>`_.
|
|
|
|
Recommended Dependencies
|
|
---------------------------------
|
|
|
|
To run examples and begin development, we recommend installing:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install torch jupyter
|
|
|
|
Recommended Python environment variables for jupyter notebooks
|
|
--------------------------------------------------------------
|
|
|
|
We recommend setting the following environment variable when running jupyter notebooks.
|
|
|
|
.. code-block:: bash
|
|
|
|
export PYTHONUNBUFFERED=1
|