CuPy 是 NumPy 和 SciPy 的 GPU 加速替代库,Python 开发者可用它直接在 NVIDIA CUDA 或 AMD ROCm 平台运行现有科学计算代码,用法几乎完全一致(例如
import cupy as cp 后调用 cp.arange、cp.sum 等 API)。同时它提供底层 CUDA 功能,支持 RawKernels、Stream 以及直接调用 CUDA Runtime API。安装方面,CuPy 通过 PyPI 提供 Linux 和 Windows 的二进制包(含
cupy-cuda12x、cupy-cuda13x 及实验性的 cupy-rocm-7-0),也有 Conda-Forge 包和 Docker 镜像。项目采用 MIT 协议,由 Preferred Networks 和社区贡献者维护。GitHub
#GitHub #开源 #CuPy #GPU #NumPy #SciPy #CUDA #ROCm #Python #科学计算
📢 频道:@GitHubTrendingHub
