https://github.com/nvidia/cuda-python

CUDA Python: Performance meets Productivity
https://github.com/nvidia/cuda-python

Last synced: 1 day 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 (over 3 years ago)
• Default Branch: main
• Last Pushed: 2024-12-20T19:29:47.000Z (13 days ago)
• Last Synced: 2024-12-25T15:06:16.016Z (8 days ago)
• Language: Python
• Homepage: https://nvidia.github.io/cuda-python/
• Size: 24 MB
• Stars: 1,013
• Watchers: 36
• Forks: 83
• Open Issues: 108
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE

Awesome Lists containing this project

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/): Pythonic exposure of CUB cooperative algorithms

* [cuda.parallel](https://nvidia.github.io/cccl/cuda_parallel/): Pythonic exposure of Thrust parallel algorithms

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