当tensorflow模型太大时系统自动重启

Question

我正在使用nvidia GTX1080 gpu(8GB)在tensorflow 上运行Inception模型,当我设置batch_size = 16和image_size = 400时,然后在我启动程序后,我的ubuntu14.04将自动重启.

Answer 1

确保不是电源装置问题。我在我的开发机器上观察到奇怪的偶尔重新启动。当我增加输入的大小（批量大小、更大的神经网络）时，重新启动的速度也在增加。原来是PSU的问题。快速检查是限制 GPU 功耗并查看此行为是否会消失。例如，您可以使用此命令将功率限制为大约 150 瓦（您需要 sudo 权限）：

sudo nvidia-smi -pl 150