tensorflow - 如何使用 tf.Dataset 将数据加载到多个 GPU 中？

Question

目前，我正在使用 get_next() 将数据传递到多个 GPU。有没有更好的方法将数据馈送到多个 GPU 中？

Answer 1

看看针对分布式训练的分片类方法。这可能是你需要的。