python - 如何使用我的 GPU 加速网络训练？

Question

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

Answer 1

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

如果您还没有，您可能想尝试在 PyPy 下运行您的训练器——对于某些工作负载，它比 CPython 快得多。也许这是其中的工作负载之一。:)

Answer 2

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

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