CuPy

CuPy 是一个用于 Python GPU 加速计算的开源数组库。CuPy 利用 CUDA Toolkit 库（包括 cuBLAS、cuRAND、cuSOLVER、cuSPARSE、cuFFT、cuDNN 和 NCCL）充分发挥 GPU 架构的性能。 该图展示了 CuPy 相对于 NumPy 的加速效果。大多数操作在使用 CuPy 时无需额外配置即可在 GPU 上良好运行，部分操作速度提升超过 100 倍。 CuPy 的接口与 NumPy 和 SciPy 高度兼容；在大多数情况下可直接替换使用。只需将 Python 代码中的 numpy 和 scipy 替换为 cupy 和 cupyx.scipy 即可。学习 CuPy 的入门教程有助于快速上手。 CuPy 支持多种方法、索引、数据类型、广播等功能。下表列出了 NumPy / SciPy API 及其对应的 CuPy 实现。