如何检查我在Tensorflow2.0上的模型训练是否使用GPU加速

Question

我已经使用fashion_mnist数据集成功地在tensorflow 2.0中运行了训练模型，并且想知道如何才能知道是否在gpu加速的帮助下训练了模型？

Answer 1

在TensorFlow 2.0中，您可以使用tf.config.experimental.list_physical_devices('GPU')：

import tensorflow as tf
gpu = tf.config.experimental.list_physical_devices('GPU')
   print("Name:", gpu.name, "  Type:", gpu.device_type)

如果您安装了GPU，它将输出以下内容：

Name: /physical_device:GPU:0   Type: GPU

并使用以下代码检查您正在使用的GPU：

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