CuPy
CuPy is an open-source array library for GPU-accelerated computing with Python. CuPy utilizes CUDA Toolkit libraries including cuBLAS, cuRAND, cuSOLVER, cuSPARSE, cuFFT, cuDNN and NCCL to make full use of the GPU architecture.
Features
- Ad-free
- Python-based
Tags
- array
- python-lib
- python-library
- Python
CuPy News & Activities
Recent activities
CuPy information
Developed by
CuPy Licensing
Open Source (MIT) and Free product.Written in
Alternatives
52 alternatives listedSupported Languages
- English
AlternativeTo Categories
Development, Education & ReferenceGitHub repository
- 10,723 Stars
- 982 Forks
- 618 Open Issues
- Updated
What is CuPy?
CuPy is an open-source array library for GPU-accelerated computing with Python. CuPy utilizes CUDA Toolkit libraries including cuBLAS, cuRAND, cuSOLVER, cuSPARSE, cuFFT, cuDNN and NCCL to make full use of the GPU architecture. The figure shows CuPy speedup over NumPy. Most operations perform well on a GPU using CuPy out of the box. CuPy speeds up some operations more than 100X.
CuPy's interface is highly compatible with NumPy and SciPy; in most cases it can be used as a drop-in replacement. All you need to do is just replace numpy and scipy with cupy and cupyx.scipy in your Python code. The Basics of CuPy tutorial is useful to learn first steps with CuPy. CuPy supports various methods, indexing, data types, broadcasting and more. This comparison table shows a list of NumPy / SciPy APIs and their corresponding CuPy implementations.