CuPy is an open source implementation of NumPy-compatible multi-dimensional array accelerated with NVIDIA CUDA. It consists of cupy.ndarray, a core multi-dimensional array class and many functions on it.
CuPy offers GPU accelerated computing with Python, using CUDA-related libraries to fully utilize the GPU architecture. According to benchmarks, it can even speed up some operations by more than 100X. CuPy is highly compatible with NumPy, serving as a drop-in replacement in most cases.
CuPy is very easy to install through pip or through precompiled binary packages called wheels for recommended environments. It also makes writing a custom CUDA kernel very easy, requiring only a small code snippet of C++.
Features
- GPU accelerated computing with Python
- Highly compatible with NumPy
- Easy installation
- Easy creation of a custom CUDA kernel
Categories
LibrariesLicense
MIT LicenseOther Useful Business Software
Ship Agents Faster
Gemini Enterprise Agent Platform lets you rapidly build, scale, govern and optimize production-ready agents grounded in your organization's data. The platform enables developers to build custom or pre-built agents for virtually any use case. New customers get $300 in free credits.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of CuPy!