Skip to content

kmatsuura/cupy

 
 

Repository files navigation

CuPy : NumPy-like API accelerated with CUDA

pypi GitHub license travis coveralls Read the Docs

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

License

MIT License (see LICENSE file).

About

NumPy-like API accelerated with CUDA

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 87.7%
  • C++ 8.7%
  • C 2.4%
  • Cuda 1.2%