GitHub - cupy/cupy: NumPy-like API accelerated with CUDA
source link: https://github.com/cupy/cupy
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
CuPy : A NumPy-compatible array library accelerated by CUDA
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"
}
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK