使用 tensorflow.slim.learning 进行多 GPU 训练

【问题标题】：Multiple GPU training with tensorfow.slim.learning使用 tensorflow.slim.learning 进行多 GPU 训练
【发布时间】：2016-10-20 01:14:02
【问题描述】：

tensorflow.contrib.slim 中的新学习模块看起来很有前途： https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/slim/python/slim/learning.py 我试图弄清楚如何在我只有一个工作节点但上面有多个 GPU 的配置上使用这个新模块重现 CIFAR 10 多 GPU 示例（或 ImageNet 示例）。

【问题讨论】：

【解决方案1】：

我使用https://github.com/tensorflow/models/tree/master/slim/deployment 取得了一些成功创建配置对象时，您将设置 num_clones = [num_gpus]

例如，

config = model_deploy.DeploymentConfig(num_clones=2)

查看 model_deploy.py 文件中的示例。

【讨论】：