CuPy – NumPy-compatible matrix library accelerated by CUDA

CuPy – NumPy-compatible matrix library accelerated by CUDA

The easiest way to install CuPy is to use pip. CuPy provides Wheels (precompiled binary packages) for the recommended environments. The install script in the source code automatically detects installed versions of CUDA, cuDNN and NCCL in your environment.

Source: cupy.chainer.org