NumPy 会自动检测并利用 GPU 吗?
2024-09-01 21:53:59
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.
小结:
- NumPy 不会主动检测并利用 GPU;
- 在安装 NumPy 的时候,为了速度更快,我们可以安装带 mkl 的版本,尽管这仍没有利用 GPU;
- 使用 GPU 版本的 NumPy 替代品,如 PyTorch。
References
PyTorch
Does Numpy automatically detect and use GPU?
NumPy GPU acceleration - Scott Sievert
最新文章
- C语言程序设计第5堂作业
- Fiddler 教程
- [LintCode] Wiggle Sort II 扭动排序之二
- 【IIS8】在IIS8添加WCF服务支持
- JS 面向对象 编程设计
- Eclipse 中Tomcat 启动 与直接启动Tomcat的区别
- [Ubuntu] Ubuntu DNS服务器配置
- Slim - 超轻量级PHP Restful API构建框架
- Java基础知识强化105:打印数组的方法总结
- POJ2225+BFS
- Steve Yegge:Google面试秘籍
- VS2010或2012中,如何设置代码格式化?
- php代码中的细节问题
- 结合apache安装subversion
- ansible字符串的处理
- 44-0-STM32的CAN外设
- java中的getStackTrace和printStackTrace的区别
- SQL server 获得 表的主键,自增键
- Kafka Docker集群搭建
- ubuntu下同时安装anaconda2与anaconda3,并分别安装与之对应的软件