CuPy 是一个用于 Python 的开源数组库，支持 GPU 加速计算。CuPy 利用 CUDA 工具包中的 cuBLAS、cuRAND、cuSOLVER、cuSPARSE、cuFFT、cuDNN 和 NCCL 等库，充分挖掘 GPU 架构的性能。

图中展示了 CuPy 相对于 NumPy 的加速效果。大多数操作在使用 CuPy 时，无需额外配置即可在 GPU 上高效运行，部分操作的加速性能可达 100 倍以上。

CuPy 的接口与 NumPy 和 SciPy 高度兼容，大多数情况下可直接作为替代方案使用。你只需将代码中的 numpy 和 scipy 替换为 cupy 和 cupyx.scipy 即可。《CuPy 基础教程》有助于你快速入门 CuPy。

CuPy 支持多种方法、索引、数据类型、广播等特性。下表列出了 NumPy/SciPy 的 API 以及其对应的 CuPy 实现。