gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-03 13:40:35 25 4
gpt4 key购买 nike

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

最佳答案

确保它不是电源单元问题。我在我的开发机器上观察到奇怪的偶尔重启。随着我增加输入的大小(批量大小、更大的神经网络),重启率也在增加。原来是电源问题。快速检查是限制 GPU 功耗,看看这种行为是否会消失。例如,您可以使用此命令将功率限制在 150 瓦左右(您需要 sudo 权限):

sudo nvidia-smi -pl 150

关于tensorflow - 当 tensorflow 模型太大时系统自动重启,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39122984/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com