检测tensorflow是否可以使用GPU

方法1

import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

方法2

import tensorflow as tf

tf = tf.Session(config=tf.ConfigProto(log_device_placement=True))
tf.list_devices()

方法3

import tensorflow as tf

tf.test.gpu_device_name()

方法4

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