丘比

CuPy是一个开源矩阵库,由NVIDIA CUDA加速。它还使用CUDA相关的库,包括cuBLAS、cuDNN、cuRand、cuSolver、cuSPARSE、cuFFT和NCCL来充分利用GPU架构。CuPy的接口与NumPy高度兼容;在大多数情况下,它可以作为一个插入式的替代品。您只需在Python代码中将numpy替换为cupy。它支持各种方法、索引、数据类型、广播等。