TensorFlow:简单的tf.learn多个GPU之间的图形复制示例

时间:2017-10-17 21:13:29

标签: tensorflow

我正在对已有的TF单GPU解决方案进行脑部手术,使其适用于多个GPU。

我已经问过Simple TensorFlow example loading one copy of a model onto each GPU that's available了(感谢Yaroslav!)但是我无法调整当前代码来获得在复制模型中进行传播的对帐。

Distributed tensorflow: the difference between In-graph replication and Between-graph replication我看到@mmry的回答:

  

“图中复制”是我们在TensorFlow中尝试的第一种方法,它没有达到许多用户所需的性能,因此更复杂的“图之间”方法是当前推荐的分布式培训方法。诸如tf.learn之类的高级库使用“图之间”方法进行分布式培训。

在多GPU配置中,哪里可以找到使用单个模型副本的tf.learn的简单示例?

我发现这也是一个github问题:https://github.com/tflearn/tflearn/issues/696

0 个答案:

没有答案