我一直在尝试使用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上运行所有程序
我在做什么错了?
答案 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
…为我工作。