Armadillo(+ BLAS)使用GPU

Mil*_*zet 6 gpu blas armadillo

是否可以使用GPU运行armadillos计算?有没有办法使用GPU blas库(例如cuBLAS)与犰狳?请注意,我对GPU编程完全不熟悉.

小智 21

目前接受的答案已经过时.与CUDA 6(目前Release Candidate状态)一起,有一个名为NVBLAS的真正替代品,它负责GPU交互,并与犰狳结合使用.您可以通过再次链接程序libnvblas.so来使用NVBLAS

但是,并非所有BLAS方法都可用,因此您必须指定一个fallback-BLAS库(如openblas).

有关更多详细信息,请参阅https://developer.nvidia.com/cublasxt


Dir*_*tel 0

不它不是。

您不能获取cpu的代码并假设它在GPU上运行,因为两者具有完全不同的内存模型、硬件、编程风格等。