LinxVIEW论坛-斯科道

标题: CuPy是使用NVIDIA CUDA加速的开源矩阵库 [打印本页]

作者: Scadao    时间: 2019-12-14 16:13
标题: CuPy是使用NVIDIA CUDA加速的开源矩阵库
CuPy使用Python提供GPU加速计算。 CuPy使用与CUDA相关的库(包括cuBLAS,cuDNN,cuRand,cuSolver,cuSPARSE,cuFFT和NCCL)来充分利用GPU架构。


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





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









欢迎光临 LinxVIEW论坛-斯科道 (http://linxview.com/) Powered by Discuz! X3.2