cupy

NumPy & SciPy for GPU

https://github.com/cupy/cupy

Science Score: 54.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
    Found CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
    Found .zenodo.json file
  • DOI references
  • Academic publication links
  • Committers with academic emails
    33 of 399 committers (8.3%) from academic institutions
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (11.6%) to scientific vocabulary

Keywords

cublas cuda cudnn cupy curand cusolver cusparse cusparselt cutensor gpu nccl numpy nvrtc nvtx python rocm scipy tensor

Keywords from Contributors

closember optimizing-compiler jax automatic-differentiation cryptocurrencies tensors symbolic-computation term-rewriting-system theano aesara
Last synced: 6 months ago · JSON representation ·

Repository

NumPy & SciPy for GPU

Basic Info
  • Host: GitHub
  • Owner: cupy
  • License: mit
  • Language: Python
  • Default Branch: main
  • Homepage: https://cupy.dev
  • Size: 43.7 MB
Statistics
  • Stars: 10,462
  • Watchers: 128
  • Forks: 948
  • Open Issues: 623
  • Releases: 0
Topics
cublas cuda cudnn cupy curand cusolver cusparse cusparselt cutensor gpu nccl numpy nvrtc nvtx python rocm scipy tensor
Created over 9 years ago · Last pushed 6 months ago
Metadata Files
Readme Contributing License Code of conduct Citation Codeowners

README.md

CuPy : NumPy & SciPy for GPU

pypi Conda GitHub license Matrix Twitter Medium

Website | Install | Tutorial | Examples | Documentation | API Reference | Forum

CuPy is a NumPy/SciPy-compatible array library for GPU-accelerated computing with Python. CuPy acts as a drop-in replacement to run existing NumPy/SciPy code on NVIDIA CUDA or AMD ROCm platforms.

```py

import cupy as cp x = cp.arange(6).reshape(2, 3).astype('f') x array([[ 0., 1., 2.], [ 3., 4., 5.]], dtype=float32) x.sum(axis=1) array([ 3., 12.], dtype=float32) ```

CuPy also provides access to low-level CUDA features. You can pass ndarray to existing CUDA C/C++ programs via RawKernels, use Streams for performance, or even call CUDA Runtime APIs directly.

Installation

Pip

Binary packages (wheels) are available for Linux and Windows on PyPI. Choose the right package for your platform.

| Platform | Architecture | Command | | --------------------- | ----------------- | ------------------------------------------------------------- | | CUDA 11.x (11.2+) | x8664 / aarch64 | pip install cupy-cuda11x | | CUDA 12.x | x8664 / aarch64 | pip install cupy-cuda12x | | CUDA 13.x | x8664 / aarch64 | pip install cupy-cuda13x | | ROCm 4.3 (experimental) | x8664 | pip install cupy-rocm-4-3 | | ROCm 5.0 (experimental) | x86_64 | pip install cupy-rocm-5-0 |

[!NOTE]\ To install pre-releases, append --pre -U -f https://pip.cupy.dev/pre (e.g., pip install cupy-cuda11x --pre -U -f https://pip.cupy.dev/pre).

Conda

Binary packages are also available for Linux and Windows on Conda-Forge.

| Platform | Architecture | Command | | --------------------- | --------------------------- | ------------------------------------------------------------- | | CUDA | x86_64 / aarch64 / ppc64le | conda install -c conda-forge cupy |

If you need a slim installation (without also getting CUDA dependencies installed), you can do conda install -c conda-forge cupy-core.

If you need to use a particular CUDA version (say 12.0), you can use the cuda-version metapackage to select the version, e.g. conda install -c conda-forge cupy cuda-version=12.0.

[!NOTE]\ If you encounter any problem with CuPy installed from conda-forge, please feel free to report to cupy-feedstock, and we will help investigate if it is just a packaging issue in conda-forge's recipe or a real issue in CuPy.

Docker

Use NVIDIA Container Toolkit to run CuPy container images.

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

Resources

[^1]: cuSignal is now part of CuPy starting v13.0.0.

License

MIT License (see LICENSE file).

CuPy is designed based on NumPy's API and SciPy's API (see docs/source/license.rst file).

CuPy is being developed and maintained by Preferred Networks 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). [PDF]

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

Owner

  • Name: CuPy
  • Login: cupy
  • Kind: organization

NumPy & SciPy for GPU

Citation (CITATION.bib)

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

Committers

Last synced: 8 months ago

All Time
  • Total Commits: 22,799
  • Total Committers: 399
  • Avg Commits per committer: 57.14
  • Development Distribution Score (DDS): 0.901
Past Year
  • Commits: 920
  • Committers: 40
  • Avg Commits per committer: 23.0
  • Development Distribution Score (DDS): 0.76
Top Committers
Name Email Commits
Kenichi Maehashi w****r@k****m 2,261
Leo Fang l****g@b****v 2,145
Yuya Unno u****o@g****m 1,992
Ryosuke Okuta o****a@p****p 1,610
Masayuki Takagi k****a@g****m 1,409
Akifumi Imanishi i****i@p****p 1,143
Kenta OONO o****o@p****p 905
Seiya Tokui b****b@g****m 880
Toshiki Kataoka k****a@p****p 765
Emilio Castillo e****l@p****p 682
Akira Naruse a****e@n****m 670
Gregory Lee g****7@g****m 582
Edgar Andrés Margffoy Tuay a****y@g****m 440
niboshi n****0@g****m 396
Evgeni Burovski e****y@g****m 304
Corey J. Nolet c****t@g****m 276
Yusuke Niitani y****i@g****m 254
Shunta Saito s****o@g****m 245
Dahlia-Chehata d****a@g****m 240
fukatani n****a@g****m 231
Masaki Saito m****o@p****p 187
Piyush-555 p****5@g****m 186
Khushi Agrawal k****1@g****m 175
SS 6****a 174
Aaron Meurer a****r@g****m 168
yoshikawa y****a@p****p 167
Gentaro Watanabe g****b@p****p 154
setoyama k****a@g****m 142
Toru Ogawa H****e 121
Naoto Mizuno m****o@e****p 113
and 369 more...

Issues and Pull Requests

Last synced: 6 months ago

All Time
  • Total issues: 769
  • Total pull requests: 1,920
  • Average time to close issues: 12 months
  • Average time to close pull requests: about 1 month
  • Total issue authors: 380
  • Total pull request authors: 143
  • Average comments per issue: 2.63
  • Average comments per pull request: 2.86
  • Merged pull requests: 1,461
  • Bot issues: 0
  • Bot pull requests: 0
Past Year
  • Issues: 247
  • Pull requests: 852
  • Average time to close issues: 12 days
  • Average time to close pull requests: 5 days
  • Issue authors: 133
  • Pull request authors: 62
  • Average comments per issue: 1.2
  • Average comments per pull request: 1.99
  • Merged pull requests: 628
  • Bot issues: 0
  • Bot pull requests: 0
Top Authors
Issue Authors
  • kmaehashi (121)
  • leofang (50)
  • AnonymousPlayer2000 (36)
  • asi1024 (16)
  • ev-br (13)
  • takagi (13)
  • jakirkham (10)
  • mrakgr (9)
  • EarlMilktea (8)
  • apiqwe (8)
  • niboshi (8)
  • sambaPython24 (8)
  • martinResearch (7)
  • siddmittal (6)
  • bmerry (5)
Pull Request Authors
  • chainer-ci (462)
  • kmaehashi (411)
  • asi1024 (311)
  • EarlMilktea (82)
  • andfoy (69)
  • leofang (65)
  • takagi (64)
  • ev-br (59)
  • grlee77 (26)
  • emcastillo (25)
  • jakirkham (22)
  • seberg (16)
  • jemiryguo (14)
  • steppi (12)
  • LaterSpec (9)
Top Labels
Issue Labels
cat:bug (255) issue-checked (160) cat:feature (111) prio:medium (103) prio:high (93) cat:enhancement (79) contribution welcome (78) prio:low (36) cat:document (34) pr-ongoing (24) cat:test (21) to-be-backported (15) st:needs-discussion (12) cat:performance (12) cat:install (10) good first issue (10) blocking (8) cat:code-fix (8) hip (7) no-compat (5) st:awaiting-author (4) cat:numpy-compat (4) np2-v13 (4) delete (4)
Pull Request Labels
prio:medium (835) prio:high (659) to-be-backported (543) backport (534) cat:test (367) cat:enhancement (305) cat:bug (283) blocking (236) cat:feature (183) cat:numpy-compat (137) cat:document (137) cat:install (88) cat:code-fix (70) cat:other (65) skip-ci (65) no-compat (43) cat:performance (20) prio:low (19) st:awaiting-author (14) hip (5) takeover (4) array-api (3) st:needs-discussion (2) st:test-and-merge (1) cat:example (1)

Packages

  • Total packages: 26
  • Total downloads:
    • pypi 3,142,922 last-month
  • Total docker downloads: 1,333,084
  • Total dependent packages: 294
    (may contain duplicates)
  • Total dependent repositories: 1,451
    (may contain duplicates)
  • Total versions: 752
  • Total maintainers: 10
pypi.org: cupy

CuPy: NumPy & SciPy for GPU

  • Versions: 147
  • Dependent Packages: 91
  • Dependent Repositories: 495
  • Downloads: 32,142 Last month
  • Docker Downloads: 1,161,198
Rankings
Dependent packages count: 0.3%
Stargazers count: 0.3%
Dependent repos count: 0.7%
Docker downloads count: 0.8%
Average: 1.0%
Forks count: 1.6%
Downloads: 2.1%
Last synced: 6 months ago
pypi.org: cupy-cuda11x

CuPy: NumPy & SciPy for GPU

  • Versions: 20
  • Dependent Packages: 44
  • Dependent Repositories: 118
  • Downloads: 285,790 Last month
  • Docker Downloads: 2,152
Rankings
Stargazers count: 0.3%
Dependent packages count: 0.6%
Downloads: 1.1%
Average: 1.2%
Dependent repos count: 1.4%
Forks count: 1.6%
Docker downloads count: 2.3%
Last synced: 6 months ago
pypi.org: cupy-cuda111

CuPy: NumPy & SciPy for GPU

  • Versions: 30
  • Dependent Packages: 12
  • Dependent Repositories: 76
  • Downloads: 1,622 Last month
  • Docker Downloads: 3,072
Rankings
Stargazers count: 0.3%
Dependent packages count: 0.7%
Forks count: 1.6%
Dependent repos count: 1.7%
Average: 1.8%
Docker downloads count: 2.0%
Downloads: 4.2%
Maintainers (4)
Last synced: 6 months ago
pypi.org: cupy-cuda110

CuPy: NumPy & SciPy for GPU

  • Versions: 32
  • Dependent Packages: 12
  • Dependent Repositories: 36
  • Downloads: 1,860 Last month
  • Docker Downloads: 297
Rankings
Stargazers count: 0.3%
Dependent packages count: 0.7%
Forks count: 1.6%
Average: 1.8%
Docker downloads count: 2.2%
Dependent repos count: 2.5%
Downloads: 3.3%
Maintainers (5)
Last synced: 6 months ago
pypi.org: cupy-cuda100

CuPy: NumPy & SciPy for GPU

  • Versions: 36
  • Dependent Packages: 7
  • Dependent Repositories: 205
  • Downloads: 1,187 Last month
Rankings
Stargazers count: 0.3%
Dependent repos count: 1.1%
Dependent packages count: 1.3%
Forks count: 1.6%
Average: 1.8%
Downloads: 4.6%
Maintainers (6)
Last synced: 6 months ago
pypi.org: cupy-cuda102

CuPy: NumPy & SciPy for GPU

  • Versions: 40
  • Dependent Packages: 11
  • Dependent Repositories: 58
  • Downloads: 2,984 Last month
  • Docker Downloads: 11
Rankings
Stargazers count: 0.3%
Dependent packages count: 0.9%
Forks count: 1.6%
Dependent repos count: 1.9%
Average: 2.1%
Downloads: 3.7%
Docker downloads count: 4.3%
Maintainers (5)
Last synced: 6 months ago
pypi.org: cupy-cuda113

CuPy: NumPy & SciPy for GPU

  • Versions: 13
  • Dependent Packages: 6
  • Dependent Repositories: 112
  • Downloads: 1,394 Last month
  • Docker Downloads: 483
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.4%
Dependent repos count: 1.4%
Forks count: 1.6%
Average: 2.2%
Docker downloads count: 3.1%
Downloads: 5.2%
Maintainers (2)
Last synced: 6 months ago
pypi.org: cupy-cuda114

CuPy: NumPy & SciPy for GPU

  • Versions: 12
  • Dependent Packages: 5
  • Dependent Repositories: 21
  • Downloads: 1,470 Last month
  • Docker Downloads: 1,053
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.6%
Forks count: 1.6%
Docker downloads count: 1.8%
Average: 2.2%
Dependent repos count: 3.2%
Downloads: 4.8%
Maintainers (3)
Last synced: 6 months ago
pypi.org: cupy-cuda112

CuPy: NumPy & SciPy for GPU

  • Versions: 17
  • Dependent Packages: 9
  • Dependent Repositories: 22
  • Downloads: 1,973 Last month
  • Docker Downloads: 44,009
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.0%
Forks count: 1.6%
Average: 2.3%
Docker downloads count: 2.7%
Dependent repos count: 3.1%
Downloads: 4.9%
Maintainers (3)
Last synced: 6 months ago
pypi.org: cupy-cuda12x

CuPy: NumPy & SciPy for GPU

  • Versions: 15
  • Dependent Packages: 45
  • Dependent Repositories: 4
  • Downloads: 2,789,499 Last month
  • Docker Downloads: 110,089
Rankings
Stargazers count: 0.3%
Dependent packages count: 0.7%
Forks count: 1.6%
Downloads: 1.8%
Docker downloads count: 2.2%
Average: 2.4%
Dependent repos count: 7.5%
Maintainers (4)
Last synced: 6 months ago
pypi.org: cupy-cuda101

CuPy: NumPy & SciPy for GPU

  • Versions: 33
  • Dependent Packages: 7
  • Dependent Repositories: 107
  • Downloads: 1,329 Last month
  • Docker Downloads: 11
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.3%
Dependent repos count: 1.4%
Forks count: 1.6%
Average: 2.5%
Docker downloads count: 4.0%
Downloads: 6.4%
Last synced: 6 months ago
pypi.org: cupy-cuda117

CuPy: NumPy & SciPy for GPU

  • Versions: 1
  • Dependent Packages: 6
  • Dependent Repositories: 10
  • Downloads: 653 Last month
  • Docker Downloads: 6,530
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.4%
Forks count: 1.6%
Average: 2.8%
Docker downloads count: 3.0%
Dependent repos count: 4.6%
Downloads: 5.9%
Maintainers (3)
Last synced: 6 months ago
pypi.org: cupy-cuda116

CuPy: NumPy & SciPy for GPU

  • Versions: 5
  • Dependent Packages: 4
  • Dependent Repositories: 20
  • Downloads: 2,558 Last month
  • Docker Downloads: 138
Rankings
Stargazers count: 0.3%
Forks count: 1.6%
Dependent packages count: 1.9%
Average: 2.9%
Dependent repos count: 3.2%
Docker downloads count: 3.9%
Downloads: 6.1%
Maintainers (3)
Last synced: 6 months ago
pypi.org: cupy-cuda90

CuPy: A NumPy-compatible array library accelerated by CUDA

  • Versions: 37
  • Dependent Packages: 5
  • Dependent Repositories: 21
  • Downloads: 759 Last month
  • Docker Downloads: 496
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.6%
Forks count: 1.6%
Docker downloads count: 1.8%
Average: 3.2%
Dependent repos count: 3.2%
Downloads: 10.4%
Last synced: 6 months ago
pypi.org: cupy-cuda115

CuPy: NumPy & SciPy for GPU

  • Versions: 9
  • Dependent Packages: 4
  • Dependent Repositories: 20
  • Downloads: 582 Last month
  • Docker Downloads: 3,508
Rankings
Stargazers count: 0.3%
Forks count: 1.6%
Docker downloads count: 1.7%
Dependent packages count: 1.9%
Average: 3.2%
Dependent repos count: 3.2%
Downloads: 10.4%
Maintainers (3)
Last synced: 6 months ago
pypi.org: cupy-cuda92

CuPy: NumPy & SciPy for GPU

  • Versions: 42
  • Dependent Packages: 5
  • Dependent Repositories: 11
  • Downloads: 820 Last month
  • Docker Downloads: 14
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.6%
Forks count: 1.6%
Average: 3.6%
Docker downloads count: 4.3%
Dependent repos count: 4.4%
Downloads: 9.3%
Last synced: 6 months ago
pypi.org: cupy-cuda91

CuPy: NumPy-like API accelerated with CUDA

  • Versions: 30
  • Dependent Packages: 5
  • Dependent Repositories: 6
  • Downloads: 389 Last month
  • Docker Downloads: 14
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.6%
Forks count: 1.6%
Average: 4.1%
Docker downloads count: 4.3%
Dependent repos count: 6.0%
Downloads: 10.7%
Last synced: 6 months ago
pypi.org: cupy-cuda80

CuPy: NumPy-like API accelerated with CUDA

  • Versions: 30
  • Dependent Packages: 5
  • Dependent Repositories: 7
  • Downloads: 613 Last month
  • Docker Downloads: 0
Rankings
Stargazers count: 0.3%
Dependent packages count: 1.6%
Forks count: 1.6%
Docker downloads count: 1.9%
Average: 4.4%
Dependent repos count: 5.5%
Downloads: 15.6%
Last synced: 6 months ago
conda-forge.org: cupy
  • Homepage: https://cupy.dev/
  • License: MIT
  • Latest release: 11.3.0
    published over 3 years ago
  • Versions: 39
  • Dependent Packages: 7
  • Dependent Repositories: 49
Rankings
Stargazers count: 4.1%
Dependent repos count: 5.1%
Average: 5.9%
Forks count: 6.2%
Dependent packages count: 8.0%
Last synced: 6 months ago
pypi.org: cupy-rocm-5-0

CuPy: NumPy & SciPy for GPU

  • Versions: 21
  • Dependent Packages: 1
  • Dependent Repositories: 1
  • Downloads: 1,950 Last month
  • Docker Downloads: 9
Rankings
Stargazers count: 0.3%
Forks count: 1.6%
Docker downloads count: 4.1%
Dependent packages count: 4.8%
Average: 6.9%
Downloads: 8.9%
Dependent repos count: 21.6%
Maintainers (4)
Last synced: 6 months ago
pypi.org: cupy-rocm-4-3

CuPy: NumPy & SciPy for GPU

  • Versions: 27
  • Dependent Packages: 2
  • Dependent Repositories: 1
  • Downloads: 1,861 Last month
Rankings
Stargazers count: 0.3%
Forks count: 1.6%
Dependent packages count: 4.8%
Average: 7.4%
Downloads: 8.8%
Dependent repos count: 21.6%
Maintainers (4)
Last synced: 6 months ago
proxy.golang.org: github.com/cupy/cupy
  • Versions: 82
  • Dependent Packages: 0
  • Dependent Repositories: 0
Rankings
Dependent packages count: 7.0%
Average: 8.2%
Dependent repos count: 9.3%
Last synced: 6 months ago
pypi.org: cupy-rocm-4-0

CuPy: NumPy & SciPy for GPU

  • Versions: 15
  • Dependent Packages: 0
  • Dependent Repositories: 1
  • Downloads: 124 Last month
Rankings
Stargazers count: 0.3%
Forks count: 1.6%
Dependent packages count: 10.1%
Average: 10.2%
Downloads: 17.6%
Dependent repos count: 21.6%
Maintainers (3)
Last synced: 6 months ago
pypi.org: cupy-rocm-4-2

CuPy: NumPy & SciPy for GPU

  • Versions: 13
  • Dependent Packages: 0
  • Dependent Repositories: 1
  • Downloads: 101 Last month
Rankings
Stargazers count: 0.3%
Forks count: 1.6%
Dependent packages count: 10.1%
Average: 10.4%
Downloads: 18.5%
Dependent repos count: 21.6%
Maintainers (3)
Last synced: 6 months ago
anaconda.org: cupy
  • Homepage: https://cupy.dev/
  • License: MIT
  • Latest release: 8.3.0
    published about 5 years ago
  • Versions: 5
  • Dependent Packages: 1
  • Dependent Repositories: 49
Rankings
Stargazers count: 9.9%
Forks count: 13.1%
Average: 19.6%
Dependent repos count: 24.6%
Dependent packages count: 30.7%
Last synced: 6 months ago
pypi.org: cupy-cuda13x

CuPy: NumPy & SciPy for GPU

  • Versions: 1
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 11,262 Last month
Rankings
Dependent packages count: 8.7%
Average: 28.8%
Dependent repos count: 48.9%
Last synced: 6 months ago

Dependencies

.github/workflows/auto-cc.yml actions
  • peter-evans/create-or-update-comment v2 composite
  • peter-evans/find-comment v2 composite
.github/workflows/backport.yml actions
  • actions/checkout v3 composite
  • actions/setup-python v4 composite
.github/workflows/docker.yml actions
  • actions/checkout v3 composite
  • docker/build-push-action v4 composite
  • docker/login-action v2 composite
.github/workflows/flexci.yml actions
  • actions/checkout v3 composite
  • actions/setup-python v4 composite
.github/workflows/pr-project.yml actions
  • actions/github-script v6 composite
  • tibdex/github-app-token b62528385c34dbc9f38e5f4225ac829252d1ea92 composite
.github/workflows/pr-updated.yml actions
  • actions/upload-artifact v3 composite
.github/workflows/pretest.yml actions
  • actions/cache v3 composite
  • actions/checkout v3 composite
  • actions/setup-python v4 composite
.github/workflows/publish.yml actions
  • actions/checkout v3 composite
  • actions/setup-python v4 composite
  • pypa/gh-action-pypi-publish release/v1 composite
docker/python3/Dockerfile docker
  • nvidia/cuda 11.8.0-devel-ubuntu22.04 build
docker/rocm/Dockerfile docker
  • rocm/dev-ubuntu-20.04 5.0.1 build
docs/requirements.txt pypi
  • numpy ==1.24.
  • pydata_sphinx_theme ==0.11.0
  • scipy ==1.10.
  • sphinx ==5.3.0
  • sphinx-copybutton ==0.5.1
setup.py pypi
  • fastrlock >=0.5
  • numpy >=1.22,<1.27