检测tensorflow是否可以使用GPU_please install gpu version of tf

作者：Cpp五条 | 2024-02-16 17:27:59

踩

please install gpu version of tf

方法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()

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Cpp五条/article/detail/95622