TensorFlow：GPU上的图像预处理

Question

我有四张GPU卡，CPU不能对所有这些卡进行足够快的图像预处理。队列很快耗尽，大部分时间卡都空闲。有没有办法将QueueRunner放在这四张GPU卡中的一张上，以便一张卡进行预处理并形成队列，而其他三张卡实际上是在训练模型？

Answer 1

是的，它应该是可能的。您可以将预处理python函数包装在tf.py_func中，创建一个TF操作并使用with tf.device("gpu:0"):将其放入请求的gpu，在那里创建一个队列并在另一个gpus上使用dequeue op。