golang 中 GPU 上的向量加法

Question

我正在编写一个应用程序，需要每秒多次添加 5000 个长度的浮点向量。是否有可能让 GPU 执行计算，以及如何完成？我需要它在 Windows 和 Linux（后来是树莓派）上运行，所以 CUDA 是不可能的，因为我没有 Nvidia 显卡。

Answer 1

您无法从 Go 直接与 Nvidia GPU 对话。您需要使用cgo从 Go 调用 C 库。请参阅本演示文稿中的第 8 张幻灯片作为示例（另请参阅完整演讲）。

有一些 Go 包将我上面提到的 cgo 部分包装到 Go 库中。mumax就是这样的一个包。