Tensorflow GPU 利用率低于 10%

Question

我正在尝试使用我的 GPU NVIDIA GEFORCE GTX 1050 和 tensorflow 来训练神经网络。我尝试了不同神经网络的不同代码示例，结果总是 GPU 利用率为 8%，计算速度比 CPU 慢得多。在此屏幕中，您可以查看列车期间的利用率。

任何解决方案？

Answer 1

这取决于您的应用程序。GPU 利用率低的情况并不少见。尝试增加批量大小

Answer 2

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