Contribute Media
A thank you to everyone who makes this possible: Read More

CuPy: A NumPy-compatible Library for GPU

Description

Logo

CuPy : NumPy-like API accelerated with CUDA

Website | Docs | Install Guide | Tutorial | Examples (Official) | Forum

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.

Improve this page