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