TensorFlow：GPUでの画像前処理

Question

私は4枚のGPUカードを持っており、CPUはすべての画像処理に十分な高速処理をしません。待ち行列はすぐに消耗し、ほとんどの時間カードはアイドル状態です。 QueueRunnerをこれらの4つのGPUカードのいずれかに配置して、1つのカードが前処理してキューを形成する方法がありますか？

Answer 1

はい、可能です。 tf.py_funcに前処理Python関数をラップし、TF演算を作成し、with tf.device("gpu:0"):を使用して要求されたgpuに入れ、そこにキューを作成し、他のgpusでデキュー演算を使用できます。