Tensorflow没有在Ubuntu上的GPU上运行

Question

我已经安装了tensorflow-gpu软件包，以及在GPU上运行tensorflow的所有要求。

现在我想测试一下，所以我从图像再培训教程（https://www.tensorflow.org/tutorials/image_retraining）下载了retrain.py脚本并用我自己的图像运行它（一个文件夹中的马，汽车平面等等在文件夹中） ＆＃34;未知＆＃34;），但它不在GPU上运行（0％使用率），仅在CPU上运行（使用率为7％）

GPU：Nvidia GeForce GTX 1060 6GB，驱动程序版本：390.59，Cuda版本：9.0.176

CPU：Ryzen 7 1700x

操作系统：Ubuntu Server 18.04

Answer 1

0％的GPU使用率和7％的GPU使用率是可疑的。可能存在一些非计算瓶颈，例如：从远程磁盘读取图像。我会尝试做一堆大型的matmuls来测试它。

通常，您执行以下操作： 列出可用设备：How to get current available GPUs in tensorflow?。如果您看到GPU，TensorFlow会识别它。此时，请TensorFlow告诉您实际使用哪些设备进行操作：请参阅https://www.tensorflow.org/programmers_guide/using_gpu中的“记录设备放置”部分。如果你看到GPU在这里使用，一切都很好。也许您的操作太小或者存在导致GPU使用率接近零的另一个瓶颈。