Tensorflow GPU利用率低于10％

Question

我正在尝试使用具有tensorflow的gpu NVIDIA GEFORCE GTX 1050来训练神经网络。我尝试过使用不同神经网络的不同代码示例，结果计算得出的GPU利用率始终为8％，比CPU慢得多。 在此屏幕上，您可以查看火车期间的利用率。

有解决方案吗？

Answer 1

有时，GPU利用率受HDD速度限制。尤其是当训练数据不适合RAM并且包含数百万个图像时，我观察到一个时期的训练与从磁盘读取所有数据的时间完全相同。在我的情况下，使用ssd甚至nvm驱动器更改HDd会按比例将tensorflow训练速度提高5倍。

Answer 2

这取决于您的应用程序。GPU利用率低并不罕见。尝试增加批处理大小