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

最新文章

  1. C语言程序设计第5堂作业
  2. Fiddler 教程
  3. [LintCode] Wiggle Sort II 扭动排序之二
  4. 【IIS8】在IIS8添加WCF服务支持
  5. JS 面向对象 编程设计
  6. Eclipse 中Tomcat 启动 与直接启动Tomcat的区别
  7. [Ubuntu] Ubuntu DNS服务器配置
  8. Slim - 超轻量级PHP Restful API构建框架
  9. Java基础知识强化105:打印数组的方法总结
  10. POJ2225+BFS
  11. Steve Yegge:Google面试秘籍
  12. VS2010或2012中,如何设置代码格式化?
  13. php代码中的细节问题
  14. 结合apache安装subversion
  15. ansible字符串的处理
  16. 44-0-STM32的CAN外设
  17. java中的getStackTrace和printStackTrace的区别
  18. SQL server 获得 表的主键,自增键
  19. Kafka Docker集群搭建
  20. ubuntu下同时安装anaconda2与anaconda3,并分别安装与之对应的软件

热门文章

  1. Atoder-3620
  2. linux 根据端口关闭程序的管道命令
  3. rails gem ransack 之模糊搜索
  4. chrome 浏览器的使用技巧
  5. 一篇文章搞明白Integer、new Integer() 和 int 的概念与区别
  6. Oracle用户、角色、权限
  7. docker-compose编排参数详解
  8. 人生苦短,我用Python(3)
  9. nyoj 737 石子合并(区间DP)
  10. 《Java基础知识》Java字符串详解