Theano支持局域网上的多GPU

时间:2016-09-13 13:14:40

标签: neural-network gpu theano

我想知道theano是否可用于在局域网上多台计算机上分布的多GPU架构上训练神经网络。

我已经通过Theano的文档知道它可以在一台计算机上支持多GPU,但我想知道它如何扩展到LAN。

提前致谢。

1 个答案:

答案 0 :(得分:0)

您失去了GPU处理通过LAN进行通信的大部分好处。编写得不好的代码会导致GPU传输,即使在单个GPU上也会显着降低处理速度。要从群集中的GPU中获得最大收益,您应该查看Gradient Descent的并行版本,例如Hogwild!

https://people.eecs.berkeley.edu/~brecht/papers/hogwildTR.pdf