098de2a6创建于 26 天前历史提交

.. _quick_start:

Quick Start Guide

Compatibility Requirements

The CUTLASS DSL 4.8 release currently supports Linux / Windows and Python 3.10 - 3.14t only.

Both Linux and Windows x86_64 are supported. aarch64 is only supported on Linux. Additional platform support will be added in future releases.

CUTLASS DSL supports the same NVIDIA driver version as the corresponding CUDA Toolkit (CUDA Toolkit 12.9 or CUDA Toolkit 13.4). Specifically, for 12.9, the driver version must be 575.51.03 or later.

Installation

To ensure compatibility with the examples and code on GitHub, use the setup.sh file from the corresponding commit in the repository.

.. code-block:: bash

git clone https://github.com/NVIDIA/cutlass.git

For CUDA Toolkit 12.9:

./cutlass/python/CuTeDSL/setup.sh --cu12

For CUDA Toolkit 13.x:

./cutlass/python/CuTeDSL/setup.sh --cu13

If you just want to try out the last known stable release of the CUTLASS DSL (may not be compatible with the latest examples and code), run:

.. code-block:: bash

For CUDA Toolkit 12.9:

pip install nvidia-cutlass-dsl

For CUDA Toolkit 13.x:

pip install "nvidia-cutlass-dsl[cu13]"

The nvidia-cutlass-dsl wheel includes everything needed to generate GPU kernels.

To install a preview release, install from pypi.nvidia.com:

.. code-block:: bash

For CUDA Toolkit 12.9:

pip install --pre nvidia-cutlass-dsl --extra-index-url https://pypi.nvidia.com

For CUDA Toolkit 13.x:

pip install --pre "nvidia-cutlass-dsl[cu13]" --extra-index-url https://pypi.nvidia.com

To run examples and begin development, we recommend installing:

.. code-block:: bash

pip install torch jupyter mypy==1.19.1

We recommend setting the following environment variable when running jupyter notebooks.

.. code-block:: bash

export PYTHONUNBUFFERED=1