Welche Bibliothek verwenden Sie für Matrixberechnungen auf CUDA?

Question

Welche Bibliothek verwenden Sie für Matrixberechnungen auf CUDA? Oder gibt es welche? Es scheint, als ob jeder das selbst schreibt.

Für die üblichen Prozessoren verwende ich Eigen. Was ist mit GPUs?

Answer 1

Für dichte Matrixoperationen könnten Sie CUBLAS (mit dem CUDA Toolkit), Magma und CULAtools in Betracht ziehen.

Für spärliche Matrixoperationen können CUSPARSE (im Lieferumfang des CUDA Toolkit enthalten) und CUSP berücksichtigt werden.

Answer 2

Mit welchen Problemen haben Sie es zu tun?
Für Sparse-Matrix-Berechnungen CUSP ist eine ziemlich gute Bibliothek.
Für dichte Probleme Magma könnte eine bessere Passform sein.