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.