您如何对 keras 或 tensorflow 进行编程以在多个 GPU 上进行分区训练，假设您在一个拥有 8 个 GPU 的 amaozn ec2 实例中，并且您希望使用所有这些 GPU 来更快地训练，但您的代码仅适用于单个 cpu 或 GPU ?
问问题
561 次
1 回答
2
是的，可以在多个 GPU 上运行 Keras 模型。这暂时只能通过 TensorFlow 后端实现，因为 Theano 功能仍然相当新。我们正在考虑在不久的将来在 Theano 中添加对多 GPU 的支持（这应该相当简单）。
使用 TensorFlow 后端，您可以像在纯 TensorFlow 中一样实现此目的：在定义 Keras 层时使用 with tf.device(d) 范围。
最初从这里
于 2017-10-21T13:58:54.390 回答