如何在Tensorflow中使用GPU？

Question

我正在尝试使用Tensorflow-GPU，但它似乎仍在CPU上运行。我已经看到this有关如何安装Tensorflow-GPU的问题，在我尝试通过执行

进行验证之前，一切似乎都是正确的

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

当我这样做时，我得到以下输出：

[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不兼容。