CUDA Python is supported on all platforms that CUDA is supported. Specific dependencies are as follows:

• Driver: Linux (450.80.02 or later) Windows(456.38 or later)
• CUDA Toolkit 11.0 to 11.6

pip install cuda-python

conda install -c nvidia cuda-python

Building dependencies:

• cython>=0.29.24

Unit-test dependencies:

• pytest>=6.2.4
• pytest-benchmark>=3.4.1
• numpy>=1.21.1

Latest list of dependencies can be found at requirements.txt

To compile the extension in-place, run:

python setup.py build_ext --inplace

To compile for debugging the extension modules with gdb, pass the --debug argument to setup.py.

You can use

pip install -e .

to install the module as editible in your current Python environment (e.g. for testing of porting other libraries to use the binding).

conda env create -f docs_src/environment-docs.yml
conda activate cuda-python-docs

Then compile and install cuda-python following the steps above.

cd docs_src
make html
open build/html/index.html

git checkout gh-pages
cd docs_src
make html
cp -a build/html/. ../docs/