GPU nvidia不可用

Question

我在具有gpu gtx 1080的笔记本电脑上使用ubuntu 18.0.4。上次，我对其进行了测试，并通过运行以下代码test gpu来正确地处理张量流 今天，我发现tensorflow无法识别GPU，当我输入nvidia-smi时，表明GPU是off。有什么问题以及如何解决。我尝试通过运行以下代码来对其进行显式设置，但映射不起作用：

    with tf.device('/gpu:0'):
      a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
      b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print(sess.run(c))

Answer 1

运行以下代码以查看您的GPU是否可见

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

如果您只看到CPU而没有GPU，请更新您的cuda和cudnn版本以与您的张量流版本匹配。

希望这会有所帮助。

Answer 2

您是否暂停了笔记本？尝试完全重新启动，因为挂起会使GPU再次唤醒后不可用。