CuPy : NumPy & SciPy for GPU

CuPy is a NumPy/SciPy-compatible array library for GPU-accelerated computing with Python.

This is a CuPy wheel (precompiled binary) package for CUDA 12.x. You need to install CUDA Toolkit 12.x locally to use these packages. Alternatively, you can install this package together with all needed CUDA components from PyPI by passing the [ctk] tag::

$ pip install cupy-cuda12x[ctk]

If you have another version of CUDA, or want to build from source, refer to the Installation Guide for instructions.