CuPy : NumPy-like API accelerated with CUDA

Website | Docs | Install Guide | Tutorial | Examples (Official) | Forum (en, ja)

CuPy is an implementation of NumPy-compatible multi-dimensional array on CUDA. CuPy consists of the core multi-dimensional array class, cupy.ndarray, and many functions on it. It supports a subset of numpy.ndarray interface.

Installation

For detailed instructions on installing CuPy, see the installation guide.

You can install CuPy using pip:

$ pip install cupy

Note that if you want to enable CUDA, cuDNN, and/or NCCL, they need to be set up before installation of CuPy.

Run with Docker

We provide the official Docker image. Use nvidia-docker command to run CuPy image with GPU. You can login to the environment with bash, and run the Python interpreter.

$ nvidia-docker run -it cupy/cupy /bin/bash

Development

Please see the documentation for developers.

More information

• Release notes
• Projects using CuPy

License

MIT License (see LICENSE file).