gpt4 book ai didi

go - golang 中 GPU 上的向量加法

转载 作者:IT王子 更新时间:2023-10-29 01:09:23 27 4
gpt4 key购买 nike

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

最佳答案

您不能直接从 Go 与 Nvidia GPU 对话。你需要使用 cgo从 Go 调用 C 库。参见 slide #8 in this presentation举一个例子(另见 full talk )。

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

关于go - golang 中 GPU 上的向量加法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41416808/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com