【发布时间】:2013-08-01 01:26:56
【问题描述】:
是否可以使用 GPU 运行犰狳计算?有没有办法将 GPU blas 库(例如 cuBLAS)与犰狳一起使用?请注意,我对 GPU 编程完全陌生。
【问题讨论】:
是否可以使用 GPU 运行犰狳计算?有没有办法将 GPU blas 库(例如 cuBLAS)与犰狳一起使用?请注意,我对 GPU 编程完全陌生。
【问题讨论】:
当前接受的答案已过时。与 CUDA 6(目前为候选发布状态)一起,有一个名为 NVBLAS 的真正插入式替代品,它负责 GPU 交互并与犰狳结合使用。您可以通过将您的程序再次链接到 libnvblas.so 来使用 NVBLAS
但是,并非所有 BLAS 方法都可用,因此您必须指定一个备用 BLAS 库(如 openblas)。
【讨论】:
不,不是。
您不能为 cpu 获取代码并假设它在 gpu 上运行,因为两者具有完全不同的内存模型、硬件、编程风格等。
【讨论】:
我终于找到了在 Ubuntu 20.04 上使用 Cmake 和 CUDA 的快捷方式。
根据the official guidance,这似乎非常困难,但在 Ubuntu 20 上使用 apt 工作正常。
sudo apt install nvidia-cuda-toolkit
在这之后,别忘了根据documentation设置环境变量NVBLAS_CONFIG_FILE。
除了更改 CPU BLAS 库外,网站上的默认内容都可以。对我来说,libopenblas.a 不起作用,应该设置为libopenblas.so.0。
手动安装对我来说似乎很复杂,所以我只是对cmake的安装方法进行了一些调整。
cmake . -Dopenblas_LIBRARY=<PATH TO NVBLAS LIB>/libnvblas.so
cmake 项目没有什么不同,也不需要额外的编译标志,因为它们都是在编译 Armadillo 库时设置的。
如果出现任何问题,日志文件(默认为nvblas.log)将提供一些信息。可能是它无法打开 CPU BLAS 库。
【讨论】: