时间戳

首页 > 解决方案 > Tensorflow GPU 无法识别我的 GPU

tensorflow - Tensorflow GPU 无法识别我的 GPU

问题描述

我一直在尝试使用 Tensorflow GPU，但显然，Tersorflow 没有识别我的 GPU。

当我运行时：

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

作为输出，只有我的 CPU 出现。我已经检查了所有内容的所有版本，它们似乎是兼容的。我有 CUDA 10.1 和 CUDA 工具包、cuDNN 7.5 和 Tensorflow 1.13.1。我在 Ubuntu 18.xx 上运行一切

我究竟做错了什么？

标签： tensorflownvidiacudnn

解决方案

什么是输出：

from tensorflow.python.client import device_lib

device_lib.list_local_devices()

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

如果上面的代码没有列出任何 GPU（而你有一个）：

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

……为我工作。

推荐阅读