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

Question

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

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

Answer 1

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

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

Answer 2

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