https://github.com/cupy/cupy Skip to content Navigation Menu Toggle navigation Sign in * Product + Actions Automate any workflow + Packages Host and manage packages + Security Find and fix vulnerabilities + Codespaces Instant dev environments + GitHub Copilot Write better code with AI + Code review Manage code changes + Issues Plan and track work + Discussions Collaborate outside of code Explore + All features + Documentation + GitHub Skills + Blog * Solutions By size + Enterprise + Teams + Startups By industry + Healthcare + Financial services + Manufacturing By use case + CI/CD & Automation + DevOps + DevSecOps * Resources Topics + AI + DevOps + Security + Software Development + View all Explore + Learning Pathways + White papers, Ebooks, Webinars + Customer Stories + Partners * Open Source + GitHub Sponsors Fund open source developers + The ReadME Project GitHub community articles Repositories + Topics + Trending + Collections * Enterprise + Enterprise platform AI-powered developer platform Available add-ons + Advanced Security Enterprise-grade security features + GitHub Copilot Enterprise-grade AI features + Premium Support Enterprise-grade 24/7 support * Pricing Search or jump to... Search code, repositories, users, issues, pull requests... Search [ ] Clear Search syntax tips Provide feedback We read every piece of feedback, and take your input very seriously. [ ] [ ] Include my email address so I can be contacted Cancel Submit feedback Saved searches Use saved searches to filter your results more quickly Name [ ] Query [ ] To see all available qualifiers, see our documentation. Cancel Create saved search Sign in Sign up Reseting focus You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session. Dismiss alert {{ message }} cupy / cupy Public * * Notifications You must be signed in to change notification settings * Fork 821 * Star 8.6k NumPy & SciPy for GPU cupy.dev License MIT license 8.6k stars 821 forks Branches Tags Activity Star Notifications You must be signed in to change notification settings * Code * Issues 524 * Pull requests 82 * Actions * Projects 3 * Wiki * Security * Insights Additional navigation options * Code * Issues * Pull requests * Actions * Projects * Wiki * Security * Insights cupy/cupy This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository. main BranchesTags Go to file Code Folders and files Last commit Last Name Name message commit date Latest commit History 28,985 Commits .github .github .pfnci .pfnci cupy cupy cupy_backends cupy_backends cupyx cupyx docker docker docs docs examples examples install install tests tests third_party third_party .coveragerc .coveragerc .flake8.cython .flake8.cython .gitignore .gitignore .gitmodules .gitmodules .mergify.yml .mergify.yml .pre-commit-config.yaml .pre-commit-config.yaml .readthedocs.yaml .readthedocs.yaml CITATION.bib CITATION.bib CODE_OF_CONDUCT.md CODE_OF_CONDUCT.md LICENSE LICENSE MANIFEST.in MANIFEST.in README.md README.md codecov.yml codecov.yml setup.cfg setup.cfg setup.py setup.py View all files Repository files navigation * README * Code of conduct * MIT license [cupy_logo_1000px] 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. >>> 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+) x86_64 / aarch64 pip install cupy-cuda11x CUDA 12.x x86_64 / aarch64 pip install cupy-cuda12x ROCm 4.3 (experimental) x86_64 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 * Installation Guide - instructions on building from source * Release Notes * Projects using CuPy * Contribution Guide * GPU Acceleration in Python using CuPy and Numba (GTC November 2021 Technical Session) * GPU-Acceleration of Signal Processing Workflows using CuPy and cuSignal^1 (ICASSP'21 Tutorial) 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] @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" } Footnotes 1. cuSignal is now part of CuPy starting v13.0.0. - About NumPy & SciPy for GPU cupy.dev Topics python gpu numpy cuda cublas scipy tensor cudnn rocm cupy cusolver nccl curand cusparse nvrtc cutensor nvtx cusparselt Resources Readme License MIT license Code of conduct Code of conduct Activity Custom properties Stars 8.6k stars Watchers 127 watching Forks 821 forks Report repository Releases 142 v13.3.0 Latest Aug 22, 2024 + 141 releases Sponsor this project Sponsor Learn more about GitHub Sponsors Used by 2.1k * @fzb0316 * @MysterieuxMan * @cheritier * @arkhn * @YuHang-aw * @dsb-ifi * @QQQQQQBY * @SaidTogru + 2,049 Contributors 321 * @kmaehashi * @okuta * @unnonouno * @leofang * @takagi * @asi1024 * @delta2323 * @beam2d * @toslunar * @anaruse * @niboshi * @grlee77 * @mergify[bot] * @andfoy + 307 contributors Languages * Python 58.1% * Cython 16.5% * C++ 12.1% * C 11.5% * Cuda 1.1% * Dockerfile 0.4% * Other 0.3% Footer (c) 2024 GitHub, Inc. Footer navigation * Terms * Privacy * Security * Status * Docs * Contact * Manage cookies * Do not share my personal information You can't perform that action at this time.