cupy

by cupy

cupy / cupy

NumPy-like API accelerated with CUDA

4.6K Stars 418 Forks Last release: 2 days ago (v8.1.0) MIT License 19.3K Commits 80 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

CuPy : A NumPy-compatible array library accelerated by CUDA

pypi GitHub license travis coveralls Gitter Twitter

Website | Docs | Install Guide | Tutorial | Examples | API Reference | 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.

Installation

Wheels (precompiled binary packages) are available for Linux (Python 3.5+) and Windows (Python 3.6+). Choose the right package for your CUDA Toolkit version.

| CUDA | Command | | ----- | ------------------------------ | | v9.0 |

pip install cupy-cuda90
| | v9.2 |
pip install cupy-cuda92
| | v10.0 |
pip install cupy-cuda100
| | v10.1 |
pip install cupy-cuda101
| | v10.2 |
pip install cupy-cuda102
| | v11.0 |
pip install cupy-cuda110
| | v11.1 |
pip install cupy-cuda111
(Currently only for Windows; See #4209 for Linux) |

See the Installation Guide if you are using Conda/Anaconda or to build from source.

Run on Docker

Use NVIDIA Container Toolkit to run CuPy image with GPU.

$ docker run --gpus all -it cupy/cupy

More information

License

MIT License (see

LICENSE
file).

CuPy is designed based on NumPy's API and SciPy's API (see

docs/LICENSE_THIRD_PARTY
file).

CuPy is being maintained and developed by Preferred Networks Inc. and community contributors.

Reference

Ryosuke Okuta, Yuya Unno, Daisuke Nishino, Shohei Hido and Crissman Loomis. CuPy: A NumPy-Compatible Library for NVIDIA GPU Calculations. Proceedings of Workshop on Machine Learning Systems (LearningSys) in The Thirty-first Annual Conference on Neural Information Processing Systems (NIPS), (2017). URL

@inproceedings{cupy_learningsys2017,
  author       = "Okuta, Ryosuke and Unno, Yuya and Nishino, Daisuke and Hido, Shohei and Loomis, Crissman",
  title        = "CuPy: A NumPy-Compatible Library for NVIDIA GPU Calculations",
  booktitle    = "Proceedings of Workshop on Machine Learning Systems (LearningSys) in The Thirty-first Annual Conference on Neural Information Processing Systems (NIPS)",
  year         = "2017",
  url          = "http://learningsys.org/nips17/assets/papers/paper_16.pdf"
}

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.