Tensorflow GPU无法识别我的GPU

时间:2019-03-11 09:28:50

标签: tensorflow nvidia cudnn

我一直在尝试使用Tensorflow GPU,但显然,Tersorflow无法识别我的GPU。

当我跑步时:

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

作为输出,仅显示我的CPU。我检查了所有版本的所有版本,它们似乎兼容。我有带有CUDA Toolkit,cuDNN 7.5和Tensorflow 1.13.1的CUDA 10.1。我正在Ubuntu 18.xx上运行所有程序

我在做什么错了?

1 个答案:

答案 0 :(得分:0)

输出是什么

from tensorflow.python.client import device_lib

device_lib.list_local_devices()

在我的系统上,tensorflow无法识别 GPU ,因为它是 XLA_GPU 。我不太确定为什么XLA_GPU也不是GPU,似乎在tensorflow-gpu代码中某处缺少OR语句。

如果以上代码未列出任何GPU(并且您有一个):

pip uninstall tensorflow
pip uninstall tensorflow-gpu
pip install tensorflow-gpu 

…为我工作。