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

Question

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

Answer 1

我在一些游戏论坛上看到过类似的问题，当 GPU 负载过重时，PC 就会关闭。原因通常是 GPU 消耗的电量超出了电源单元的处理能力。检查例如这里或这里。因此，可能值得调查一下您的 PSU 是否是罪魁祸首。

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