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