CuPy: A Numpy-compatible Library for NVIDIA GPU Calculations

ML Systems Workshop in NIPS 2017

By : Ryosuke Okuta, Yuya Unno, Daisuke Nishino, Shohei Hido, Crissman Loomis


CuPy is an open-source library with NumPy syntax that increases speed by doing
matrix operations on NVIDIA GPUs. It is accelerated with the CUDA platform
from NVIDIA and also uses CUDA-related libraries, including cuBLAS, cuDNN,
cuRAND, cuSOLVER, cuSPARSE, and NCCL, to make full use of the GPU
architecture. CuPy’s interface is highly compatible with NumPy; in most cases it
can be used as a drop-in replacement. CuPy supports various methods, data types,
indexing, broadcasting, and more.

  • Twitter
  • Facebook