如何在 Tensorflow 中使用 GPU？

Question

我正在尝试使用 Tensorflow-GPU，但它似乎仍在 CPU 上运行。 我已经看到这个关于如何安装 Tensorflow-GPU 的问题，一切似乎都是正确的，直到我尝试通过执行来验证它

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

当我这样做时，我会关注 output：

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 4000308941676664793
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 7883418274751980843
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 13103704949076676658
physical_device_desc: "device: XLA_GPU device"
]

Tensorflow 似乎只找到了我的 CPU，显然是 XLA_CPU 和 XLA_GPU。 我只需要手动设置 GPU 还是我做错了什么？ 我正在使用 NVIDIA GeForce GTX 1050 Ti BTW。

Answer 1

也许您应该确认是否安装了NVIDIA驱动，cuda和cudnn及其版本

Answer 2

只有 CUDA 版本 10.1 有效。 版本 9 和 10.2 与最新版本的 Tensorflow 不兼容。