Numpy是否自动检测并使用GPU?

Numpy/Python是否自动检测GPU的存在并利用 它可以加速矩阵计算(例如numpy.multiply, 努比。利纳尔。库存。。。等等)?

不

或者我是否有特定方式的代码来利用GPU进行快速 计算

对搜索Numba、CuPy、Theano、PyTorch或PyCUDA,了解使用GPU加速Python的不同范例。

不,您也可以使用CuPy,它与numpy具有类似的接口。 https://cupy.chainer.org/

JAX使用XLA编译NumPy代码以在GPU/TPU上运行: https://github.com/google/jax

不。Numpy不使用GPU。但你可以用丘比。CuPy的语法与NumPy非常兼容。因此,要使用GPU,只需替换以下代码行

 import numpy as np

具有

import cupy as np

仅此而已。继续运行您的代码。还有一件事我想我应该在这里提到的是,要安装CuPy,首先需要安装CUDA。由于您的问题的目的是通过使用GPU来加快计算速度,因此我还建议您探索Pytork。有了PyTorch,你几乎可以用NumPy做任何事情,甚至更多。如果您已经熟悉NumPy,那么学习曲线也会非常平滑。您可以在此处找到有关使用PyTorch替换NumPy的更多详细信息: https://www.youtube.com/watch?v=p3iYN-2XL8w