https://github.com/NVIDIA/cuda-python
CUDA Python: Performance meets Productivity
https://github.com/NVIDIA/cuda-python
Last synced: 25 days ago
JSON representation
CUDA Python: Performance meets Productivity
- Host: GitHub
- URL: https://github.com/NVIDIA/cuda-python
- Owner: NVIDIA
- License: other
- Created: 2021-06-28T22:14:29.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2025-03-19T18:16:16.000Z (26 days ago)
- Last Synced: 2025-03-20T03:44:13.978Z (26 days ago)
- Language: Python
- Homepage: https://nvidia.github.io/cuda-python/
- Size: 111 MB
- Stars: 1,182
- Watchers: 36
- Forks: 100
- Open Issues: 110
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
- awesome-cuda-and-hpc - NVIDIA/cuda-python - python?style=social"/> : CUDA Python is the home for accessing NVIDIA’s CUDA platform from Python. CUDA Python Low-level Bindings. [nvidia.github.io/cuda-python/](https://nvidia.github.io/cuda-python/latest/) (Frameworks)
README
# cuda-python
CUDA Python is the home for accessing NVIDIA’s CUDA platform from Python. It consists of multiple components:
* [cuda.core](https://nvidia.github.io/cuda-python/cuda-core/latest): Pythonic access to CUDA Runtime and other core functionalities
* [cuda.bindings](https://nvidia.github.io/cuda-python/cuda-bindings/latest): Low-level Python bindings to CUDA C APIs
* [cuda.cooperative](https://nvidia.github.io/cccl/cuda_cooperative/): A Python package for easy access to highly efficient and customizable parallel algorithms, like `sort`, `scan`, `reduce`, `transform`, etc.
* [cuda.parallel](https://nvidia.github.io/cccl/cuda_parallel/): A Python package providing CUB's reusable block-wide and warp-wide primitives for use within Numba CUDA kernels
For access to NVIDIA CPU & GPU Math Libraries, please refer to [nvmath-python](https://docs.nvidia.com/cuda/nvmath-python/latest).
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](https://nvidia.github.io/cuda-python/cuda-bindings/latest) documentation for installation guide and further detail.
## cuda-python as a metapackage
`cuda-python` is being re-structured to become a metapackage that contains a collection of subpackages. Each subpackage is versioned independently, allowing installation of each component as needed.
### Subpackage: `cuda.core`
The `cuda.core` package offers idiomatic, pythonic access to CUDA Runtime and other functionalities.
The goals are to
1. Provide **idiomatic ("pythonic")** access to CUDA Driver, Runtime, and JIT compiler toolchain
2. Focus on **developer productivity** by ensuring end-to-end CUDA development can be performed quickly and entirely in Python
3. **Avoid homegrown** Python abstractions for CUDA for new Python GPU libraries starting from scratch
4. **Ease** developer **burden of maintaining** and catching up with latest CUDA features
5. **Flatten the learning curve** for current and future generations of CUDA developers
### Subpackage: `cuda.bindings`
The `cuda.bindings` package is a standard set of low-level interfaces, providing full coverage of and access to the CUDA host APIs from Python.
The list of available interfaces are:
* CUDA Driver
* CUDA Runtime
* NVRTC
* nvJitLink
## Supported Python Versions
All `cuda-python` subpackages follows CPython [End-Of-Life](https://devguide.python.org/versions/) schedule for supported Python version guarantee.
Before dropping support there will be an issue raised as a notice.