Theano:混合 CPU 和 GPU?

Theano: mixing CPU and GPU?

我构建了一个神经网络,它需要使用 Cholesky 分解并求解三角系统作为其计算的一部分。这意味着我当然还需要计算整个计算的梯度。

当我尝试编译代码时出现错误 "No Cula available"。不幸的是,我无法从 website.

下载 Cula

我想知道是否可以将 CPU 和 GPU 与 theano 混合使用。我需要使用 Cholesky 和 ​​Solve 的矩阵很小 (100x100),所以我可以在 CPU 上做那部分。不过,为此,我需要在 GPU 计算完矩阵后立即将它们传输到 CPU,然后将结果发送回 GPU。是否可以足够透明地做到这一点?

此次转账是透明的。不需要做一些特别的事情。但我会建议您分析 Theano 图,以确保:http://deeplearning.net/software/theano/tutorial/profiling.html#tut-profiling

对于cula,NVIDIA现在提供了一个基于cusolver的解决方案的PR。所以这可能不再需要了。试试这个公关:https://github.com/Theano/Theano/pull/4917