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

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