擅长:python、mysql、java
<p>问题是FIFOQueue只是CPU,所以预取只能在CPU上工作。
但是,您可以通过创建GPU变量并并行地将FIFO队列中的内容复制到GPU变量上,然后依靠这些变量而不是队列来进行GPU输入,来实现自己的预取。下面是Tim Zaman对这个想法的一个实现:</p>
<p><a href="https://github.com/tensorflow/tensorflow/issues/5722#issuecomment-266297116" rel="nofollow noreferrer">https://github.com/tensorflow/tensorflow/issues/5722#issuecomment-266297116</a></p>