yichuan9527 / cupy

NumPy-like API accelerated with CUDA

Home Page:https://cupy.chainer.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

https://cupy.chainer.org

License:Other


Languages

Language:Python 88.1%Language:C++ 8.3%Language:C 2.5%Language:Cuda 1.1%