c++ - 您使用什么库在 CUDA 上进行矩阵计算？

Question

您使用什么库在 CUDA 上进行矩阵计算？或者有吗？好像每个人都是自己写的。

对于通常的处理器，我使用Eigen。GPU 呢？

Answer 1

对于密集矩阵运算，您可以考虑CUBLAS（随 CUDA 工具包提供）、Magma和CULAtools。

对于稀疏矩阵运算，请考虑CUSPARSE（随 CUDA 工具包提供）和CUSP。

Answer 2

你正在处理什么问题？
对于稀疏矩阵计算， CUSP是一个很好的库。
对于密集问题，Magma可能更适合。