Installation¶
Runtime Requirements¶
cuda.core
is supported on all platforms that CUDA is supported. Specific
dependencies are as follows:
CUDA 11 |
CUDA 12 |
|
---|---|---|
CUDA Toolkit [1] |
11.2 - 11.8 |
12.0 - 12.6 |
Driver |
450.80.02+ (Linux), 452.39+ (Windows) |
525.60.13+ (Linux), 527.41+ (Windows) |
cuda.core
supports Python 3.9 - 3.12, on Linux (x86-64, arm64) and Windows (x86-64).
Installing from PyPI¶
cuda.core
works with cuda.bindings
(part of cuda-python
) 11 or 12. For example with CUDA 12:
$ pip install cuda-core[cu12]
and likewise use [cu11]
for CUDA 11.
Note that using cuda.core
with NVRTC or nvJitLink installed from PyPI via pip install
is currently
not supported. This will be fixed in a future release.
Installing from Source¶
$ git clone https://github.com/NVIDIA/cuda-python
$ cd cuda-python/cuda_core
$ pip install .
For now cuda-python
(cuda-bindings
later) 11.x or 12.x is a required dependency.