计算机在TensorFlow中使用大型小批量重新启动

Question

我正在使用Titan X GPU（12 GB内存）运行TensorFlow for Windows。当我尝试使用大于50张图像的迷你批次训练网络256X256X1的图像时，我的计算机只是崩溃并自动重启。使用较小的迷你批次，它运行得很好。 关于可能导致这种情况的任何线索？

Answer 1

我在一些游戏论坛上看到过类似的问题，当GPU处于高负荷状态时，PC会关闭。原因通常是GPU正在消耗的功率超过电源单元可以处理的功率。检查例如here或here。因此，可能值得调查你的PSU是否是罪魁祸首。

编辑：可能是程序SpeedFan可以帮助您调试它 - 它能够显示温度传感器的电压和读数，这也会告诉您PC是否过热（我自己从未使用过该工具） ，我也没有加入它，只是在网上找到它。