代码之家  ›  专栏  ›  技术社区  ›  Marco

TensorFlow GPU利用率低于10%

  •  1
  • Marco  · 技术社区  · 7 年前

    我正在尝试使用我的GPU Nvidia Geforce GTX 1050和TensorFlow来训练神经网络。我尝试过不同神经网络的不同代码示例,结果总是GPU利用率为8%,计算速度比CPU慢得多。 从这个屏幕上,您可以看到列车期间的利用率。

    enter image description here

    有什么解决办法吗?

    2 回复  |  直到 6 年前
        1
  •  1
  •   kerastf    7 年前

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

        2
  •  2
  •   Ivan Kovtun    6 年前

    有时GPU利用率受HDD速度的限制。尤其是当训练数据不适合RAM并且由数百万张图像组成时,我观察到一个历元训练与从磁盘读取所有数据的时间完全相同。在我的情况下,用SSD甚至NVM驱动器更换硬盘会使TensorFlow培训速度增加5倍。