Bạn sử dụng thư viện nào để tính toán ma trận trên CUDA?

Question

Bạn sử dụng thư viện nào để tính toán ma trận trên CUDA? Hoặc là có bất kỳ? Dường như mọi người tự viết điều này.

Đối với bộ xử lý thông thường, tôi sử dụng Eigen. Còn GPU thì sao?

Answer 1

Đối với các hoạt động ma trận dày đặc, bạn có thể xem xét CUBLAS (được cung cấp với Bộ công cụ CUDA), Magma và CULAtools.

Đối với các hoạt động ma trận thưa thớt, hãy xem xét CUSPARSE (được cung cấp với Bộ công cụ CUDA) và CUSP.

Answer 2

Bạn đang gặp phải vấn đề gì?
Đối với phép tính ma trận thưa thớt CUSP là một thư viện khá tốt.
Đối với các vấn đề dày đặc Magma có thể phù hợp hơn.