CuPy:CUDA加速了与NumPy兼容的数组库-开源

CuPy是NumPy兼容的多维数组的开源实现，该数组通过NVIDIA CUDA加速。 它由cupy.ndarray，一个核心的多维数组类及其上的许多函数组成。 CuPy使用与CUDA相关的库来充分利用GPU架构，从而使用Python提供GPU加速计算。 根据基准测试，它甚至可以将某些操作的速度提高100倍以上。 CuPy与NumPy高度兼容，在大多数情况下可作为即插即用的替代品。 CuPy非常容易通过pip或通过预编译的二进制程序包安装，该程序包在推荐的环境中称为wheel。 这也使编写自定义CUDA内核变得非常容易，只需要一小段C ++代码即可。