当前位置： 移动技术网 > IT编程>脚本编程>Python > NumPy 会自动检测并利用 GPU 吗？

NumPy 会自动检测并利用 GPU 吗？

2019年09月06日  | 移动技术网IT编程  | 我要评论

青苹果未删减版,伊藤弘美,启亨官网

pytorch 官网 60 分钟入门教程在介绍什么是 pytorch 时有一句话：a replacement for numpy to use the power of gpus

pytorch 是 numpy 的替代品，可以使用gpu的强大功能。难道强大的 numpy 不利用 gpu？

stack overflow 上有同样的问题：does numpy automatically detect and use gpu?
给出的回答是：numpy 不会主动检测并利用 gpu。

而为了利用 gpu，numpy 有很多替代品，像 cupy、numba，当然 pytorch 也算一种。

博客 numpy gpu acceleration - scott sievert 中做了一些实验，查看 numpy without mkl、numpy with mkl、以及 cudamat 三者在执行矩阵乘法a@b的时长：

intel® math kernel library (intel® mkl) optimizes code with minimal effort for future generations of intel® processors.

小结：

  1. numpy 不会主动检测并利用 gpu；
  2. 在安装 numpy 的时候，为了速度更快，我们可以安装带 mkl 的版本，尽管这仍没有利用 gpu；
  3. 使用 gpu 版本的 numpy 替代品，如 pytorch。

references

pytorch
does numpy automatically detect and use gpu?
numpy gpu acceleration - scott sievert

您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网