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

Question

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

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

Answer 1

如果您已经安装tensorflow-gpu并且一切正常，默认情况下，它应该在您的 GPU 上运行，除非：

• 您在代码中特别声明该图应该在 CPU 上运行，例如with tf.device('/cpu:0')，
• 您可以使用os.environ['CUDA_VISIBLE_DEVICES'] = ''.

您可以使用（归功于此答案）检查 tensorflow 是否找到了您的 GPU

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

另外，如果你有一个 NVIDIA gpu，你可以检查你的 GPU 的使用情况nvidia-smi，看看 tensorflow 确实在使用你的 GPU。

现在，如果您想分配特定的 GPU 和/或限制图形可以在 GPU 内存上分配的内存，您可能需要查看此处了解更多详细信息。