怎么查看tensorflow用的是gpu还是cpu

运行以下代码即可查看tensorflow使用的是CPU还是GPU：

import tensorflow as tf

# 创建一个tensorflow会话
sess = tf.Session()

# 获取tensorflow当前使用的设备（CPU还是GPU）
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
  print('GPU设备没有找到')
else:
  print('当前TensorFlow使用的设备：{}'.format(device_name))