tensorflow-gpu库是否在GPU上自动运行tensorflow代码（非GPU）？

Question

我是否必须自定义为非GPU Tensorflow库编写的代码以适合tensorflow-gpu库？

我有一个GPU，并且想运行只为非GPU tensorflow库编写的python代码。我可以简单地安装tensor-flow gpu模块并运行代码吗？还是为了在GPU上运行代码我需要进行任何代码更改？

Answer 1

如果您安装了DocumentIndex.class并且一切正常，则默认情况下，它应该可以在您的GPU上运行，除非：

• 您在代码中特别声明了该图应在CPU上以com/es/utility之类的方式运行，
• 您可以使用tensorflow-gpu将GPU隐藏在张量流中。

您可以使用（credit to this answer）检查tensorflow是否找到了您的GPU

with tf.device('/cpu:0')

此外，如果您拥有NVIDIA GPU，则可以使用os.environ['CUDA_VISIBLE_DEVICES'] = ''检查GPU的使用情况，并查看tensorflow确实在使用GPU。

现在，如果您想分配特定的GPU和/或限制图形可以在GPU内存上分配的内存，则可能需要看一下here，以获得更多详细信息。