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

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

网友评论