使用CPU内存来存储GPU输出

Question

我想知道是否可以告诉 cupy 将某些计算的输出（在 GPU 上进行）直接存储在 CPU 内存上。我目前正在尝试使用

cupy.scipy.sparse.eigsh

对角化一个大的稀疏矩阵，不幸的是，我遇到了内存问题，因为矩阵的本征态不稀疏并且占用了太多空间。有没有可能解决这个问题？
预先感谢。

Answer 1

有

SciPy

和

CuPy

可以做到这一点。 Scipy 在 CPU 上运行，copy 在 GPU 上运行。当你现在使用 Cupy 时，你可以尝试 Scipy： https://docs.scipy.org/doc/scipy/reference/ generated/scipy.sparse.linalg.eigsh.html