CUDA Python

CUDA Python is the home for accessing NVIDIA’s CUDA platform from Python. It consists of multiple components:

  • cuda.core: Pythonic access to CUDA runtime and other core functionalities

  • cuda.bindings: Low-level Python bindings to CUDA C APIs

  • cuda.cooperative: Pythonic exposure of CUB cooperative algorithms

  • cuda.parallel: Pythonic exposure of Thrust parallel algorithms

For access to NVIDIA Math Libraries, please refer to nvmath-python.

CUDA Python is currently undergoing an overhaul to improve existing and bring up new components. All of the previously available functionalities from the cuda-python package will continue to be available, please refer to the cuda.bindings documentation for installation guide and further detail.