Skip to content
/ cupy Public
forked from cupy/cupy

A NumPy-compatible array library accelerated by CUDA

License

Notifications You must be signed in to change notification settings

bkpcoding/cupy

 
 

Repository files navigation

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

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 and Windows. 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

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"
}

About

A NumPy-compatible array library accelerated by CUDA

Resources

License

Code of conduct

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 75.9%
  • C 12.2%
  • C++ 10.5%
  • Cuda 1.0%
  • Shell 0.2%
  • PowerShell 0.1%
  • Other 0.1%