CUDA Python
conda install nvidia::cuda-python
CUDA Python is the home for accessing NVIDIA’s CUDA platform from Python.