CUDA Python: Performance meets Productivity
Project description
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 functionality
cuda.bindings: Low-level Python bindings to CUDA C APIs
cuda.pathfinder: Utilities for locating CUDA components installed in the user’s Python environment
cuda.coop: A Python module providing CCCL’s reusable block-wide and warp-wide device primitives for use within Numba CUDA kernels
cuda.compute: A Python module for easy access to CCCL’s highly efficient and customizable parallel algorithms, like sort, scan, reduce, transform, etc. that are callable on the host
numba.cuda: A Python DSL that exposes CUDA SIMT programming model and compiles a restricted subset of Python code into CUDA kernels and device functions
cuda.tile: A new Python DSL that exposes CUDA Tile programming model and allows users to write NumPy-like code in CUDA kernels
nvmath-python: Pythonic access to NVIDIA CPU & GPU Math Libraries, with host, device, and distributed APIs. It also provides low-level Python bindings to host C APIs (nvmath.bindings).
nvshmem4py: Pythonic interface to the NVSHMEM library, enabling Python applications to leverage NVSHMEM’s high-performance PGAS (Partitioned Global Address Space) programming model for GPU-accelerated computing
Nsight Python: Python kernel profiling interface that automates performance analysis across multiple kernel configurations using NVIDIA Nsight Tools
CUPTI Python: Python APIs for creation of profiling tools that target CUDA Python applications via the CUDA Profiling Tools Interface (CUPTI)
CUDA Python is currently undergoing an overhaul to improve existing and introduce new components. All of the previously available functionality from the cuda-python package will continue to be available, please refer to the cuda.bindings documentation for installation guide and further detail.
cuda-python as a metapackage
cuda-python is being restructured 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 functionality.
The goals are to
Provide idiomatic (“pythonic”) access to CUDA Driver, Runtime, and JIT compiler toolchain
Focus on developer productivity by ensuring end-to-end CUDA development can be performed quickly and entirely in Python
Avoid homegrown Python abstractions for CUDA for new Python GPU libraries starting from scratch
Ease developer burden of maintaining and catching up with latest CUDA features
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 is:
CUDA Driver
CUDA Runtime
NVRTC
nvJitLink
NVVM
cuFile
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file cuda_python-13.1.1-py3-none-any.whl.
File metadata
- Download URL: cuda_python-13.1.1-py3-none-any.whl
- Upload date:
- Size: 8.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
944cc4fe6482673d28dd545797a28840945a1668739328fa2ad1e9be4f7050d9
|
|
| MD5 |
77977499cd7824f4e950a0511e686352
|
|
| BLAKE2b-256 |
cd08b5e3b9822662d72d540d830531e3ab6a7cabbda3dd56175696aabccfeb76
|
Provenance
The following attestation bundles were made for cuda_python-13.1.1-py3-none-any.whl:
Publisher:
release.yml on NVIDIA/cuda-python
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
cuda_python-13.1.1-py3-none-any.whl -
Subject digest:
944cc4fe6482673d28dd545797a28840945a1668739328fa2ad1e9be4f7050d9 - Sigstore transparency entry: 755683848
- Sigstore integration time:
-
Permalink:
NVIDIA/cuda-python@476cf1df490bd770011b86afdf6f5686d6c31e6c -
Branch / Tag:
refs/heads/main - Owner: https://github.com/NVIDIA
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yml@476cf1df490bd770011b86afdf6f5686d6c31e6c -
Trigger Event:
workflow_dispatch
-
Statement type: