如何使用GPU加速网络培训？

Question

我想知道是否有办法使用我的GPU来加速PyBrain中的网络培训。

Answer 1

除非PyBrain是为此设计的，否则你可能不会。

你可能想尝试在PyPy下运行你的训练师，如果你还没有 - 对于某些工作负载来说，它比CPython快得多。也许这是其中一个工作负载。 ：）

Answer 2

Checkout Pylearn2：https://github.com/lisa-lab/pylearn2

这是一个较新的库，可以通过cuda-convnet在CUDA上运行GPU。