GPU无法检测到tensorflow-gpu 2.0,问题出在哪里?为
import tensorflow as tf
print(tf.test.is_built_with_cuda())
print(tf.test.is_gpu_available())
from tensorflow.python.client import device_lib
device_lib.list_local_devices()下面是使用Ubuntu 16.04的我的电脑中的输出
True
False
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 10185800011826802770, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 2908743746130792497
physical_device_desc: "device: XLA_CPU device", name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 13796634005997726084
physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:1"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 4940094218016527544
physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:2"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 15391692776259568109
physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:3"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 8368387340417357161
physical_device_desc: "device: XLA_GPU device"]发布于 2019-10-28 11:37:22
正如y.selivonchyk指出的那样,在安装CUDA10.0之后,它现在可以工作了。虽然不容易去那里(涉及卸载/重新安装nvidia驱动程序和cuda),如果有人需要帮助,请在评论中让我知道。
https://stackoverflow.com/questions/58580781
复制相似问题