Tensorflow with GPU，如何查看tensorflow正在使用GPU？

Question

尝试安装 TensorFlow 以与 GPU 配合使用。我看到的一些文档说，当检测到时，tensorflow 会立即提供 GPU 支持。如果是这样，我可以使用什么命令来查看 TensorFlow 正在使用我的 GPU？我看过其他文档说你需要安装tensorflow-gpu。我已经尝试了这两种方法，但没有看到我的 GPU 是如何使用的？

Answer 1

转到命令行并运行 Python

以下示例列出了主机上可见 GPU 的数量。 文档

import tensorflow as tf
devices = tf.config.list_physical_devices('GPU')
print(len(devices)) 

对于 CUDA 文档

import tensorflow as tf
tf.test.is_built_with_cuda()

返回 TensorFlow 是否是使用 CUDA (GPU) 支持构建的。 文档

您也可以检查以下函数，但它已被弃用**

import tensorflow as tf

tf.test.is_gpu_available()

如果您的 GPU 可用，则两者都返回 True