CuPy is an implementation of NumPy-compatible multi-dimensional array on CUDA.
About CuPy
From the CuPy homepage:
High Performance with CUDA
CuPy is an open-source matrix library accelerated with NVIDIA CUDA. It also uses CUDA-related libraries including cuBLAS, cuDNN, cuRand, cuSolver, cuSPARSE, cuFFT and NCCL to make full use of the GPU architecture.
Highly Compatible with NumPy
CuPy's interface is highly compatible with numpy; in most cases it can be used as a drop-in replacement. All you need to do is just replace
numpy
withcupy
in your Python code. It supports various methods, indexing, data types, broadcasting and more.
CuPy consists of the core multi-dimensional array class, cupy.ndarray
, and many functions on it. It supports a subset of numpy.ndarray
interface.