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

Question

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

Answer 1

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

例如，

config = model_deploy.DeploymentConfig(num_clones=2)

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