CuPy是使用NVIDIA CUDA加速的开源矩阵库

Scadao

CuPy使用Python提供GPU加速计算。 CuPy使用与CUDA相关的库（包括cuBLAS，cuDNN，cuRand，cuSolver，cuSPARSE，cuFFT和NCCL）来充分利用GPU架构。


该图显示了CuPy在NumPy上的提速。他们中的大多数人都可以直接使用CuPy在GPU上执行良好。 CuPy将某些操作的速度提高了100倍以上。





CuPy的界面与NumPy高度兼容；在大多数情况下，它可以用作替代产品。您需要做的就是在Python代码中将numpy替换为cupy.