方法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()
本文提供了四种检测TensorFlow是否正确安装并配置了GPU支持的方法。包括使用tf.test.gpu_device_name(), tf.Session()进行设备列表展示, 以及通过device_lib.list_local_devices()获取本地设备列表。

1694

被折叠的 条评论
为什么被折叠?



