numba-cuda0.19.1
numba-cuda0.19.1
Published
CUDA target for Numba
pip install numba-cuda
Package Downloads
Authors
Project URLs
Requires Python
>=3.9
Dependencies
- numba
>=0.60.0
- cuda-bindings
<13.0.0,>=12.9.1; extra == "cu12"
- cuda-core
==0.3.*; extra == "cu12"
- cuda-python
==12.9.*; extra == "cu12"
- nvidia-cuda-nvcc-cu12
; extra == "cu12"
- nvidia-cuda-runtime-cu12
; extra == "cu12"
- nvidia-cuda-nvrtc-cu12
; extra == "cu12"
- nvidia-nvjitlink-cu12
; extra == "cu12"
- nvidia-cuda-cccl-cu12
; extra == "cu12"
- cuda-bindings
==13.*; extra == "cu13"
- cuda-core
<0.4.0dev0,==0.3.2; extra == "cu13"
- cuda-python
==13.*; extra == "cu13"
- nvidia-nvvm
==13.*; extra == "cu13"
- nvidia-cuda-runtime
==13.*; extra == "cu13"
- nvidia-cuda-nvrtc
==13.*; extra == "cu13"
- nvidia-nvjitlink
==13.*; extra == "cu13"
- nvidia-cuda-cccl
==13.*; extra == "cu13"
- pre-commit
; extra == "test"
- psutil
; extra == "test"
- cffi
; extra == "test"
- pytest
; extra == "test"
- pytest-xdist
; extra == "test"
- filecheck
; extra == "test"
- numba-cuda
[cu12]; extra == "test-cu12"
- numba-cuda
[test]; extra == "test-cu12"
- nvidia-curand-cu12
; extra == "test-cu12"
- numba-cuda
[cu13]; extra == "test-cu13"
- numba-cuda
[test]; extra == "test-cu13"
- nvidia-curand
==10.4.*; extra == "test-cu13"
Numba CUDA Target
The CUDA target for Numba. Please visit the official documentation to get started!
To report issues or file feature requests, please use the issue tracker.
To raise questions or initiate discussions, please use the Numba Discourse forum.
Installation with pip or conda
Please refer to the Installation documentation.
Installation from source
Install as an editable install:
pip install -e .
If you want to manage all run-time dependencies yourself, also pass the --no-deps
flag.
Running tests
pytest -n auto --pyargs numba.cuda.tests -v
Testing should discover the numba.cuda
module from the numba_cuda
package. You
can check where numba.cuda
files are being located by running
python -c "from numba import cuda; print(cuda.__file__)"
which will show a path like:
<path to numba-cuda repo>/numba_cuda/numba/cuda/__init__.py
Contributing Guide
Review the CONTRIBUTING.md file for information on how to contribute code and issues to the project.