Python 基于gpu的矩阵计算

今天我问我是否可以用gpu代替cpu进行矩阵计算，因为我知道gpu的设计比cpu更快。 我在网上搜索了一下，发现了关于使用不同python库的gpu进行矩阵计算的注意事项，但我的问题是有一个文档描述了我们应该如何编写代码来与gpu通信。 我这样问是因为我想开发自己的一个，以便更好地理解gpu是如何工作的，并尝试一些不同的东西。

---

感谢大家。

我用OpenCL解决了这个问题

OpenCL是GPU供应商自己实现的标准库。像NVIDIA一样，由于CUDA库，它支持openCL和其他功能

---

这里有一个很好的

你一定要看看CUDA文档，了解如何将数据从CPU传输到GPU，然后开始学习更多的抽象库来隐藏所有样板文件并只执行矩阵乘法。另一个选择是OpenCL，如果你不在NVIDIA硬件上运行，它也可以工作。