文章目录
问题可能原因解决方案
问题
前面用自己的台式机利用GPU训练模型的时候,电脑老是自动重启,当时试了各种方法,找了各种原因。
电脑配置 电脑买的是二手的,(强烈建议买新的,大牌的,后面用起来省心,不然遇到问题或者遇到以次充好的,会影响心情的)
海盗船,550w显卡:华硕1070主板:微星z270
自动重启现象
在ubuntu下,使用pytorch框架,测试torch.cuda.is_availabel(),都是可以的,但是用pytorch训练和测试模型的时候,就自动重启,基本都是程序一启动就重启,有时还会训练几代了才重启在win10下,用鲁大师进行温度压力测试的时候,一般跑到几分钟后就自动重启。
可能原因
可能有设置显卡/CPU温度保护,当高于多少度的时候,就自动重启。但是我一直没有找到。内存条,显卡的硬件问题。(所以最好不要买二手的,或者买二手的之后,一定要做好各种测试,显卡还是蛮贵的)电源功率不稳。(本人的问题就是原来的电源功率不稳。后来换了一个长城600w的,程序就跑起来了)
解决方案
买了一个600w的长城电源,京东链接,买回来之后,换上就好了。 哎,有足够预算还是买一个正规牌子的。免得自己折腾。