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

Question

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

Answer 1

看看shard类方法，该方法旨在进行分布式训练。这可能就是您所需要的。