Process finished with exit code -1073740791 (0xC0000409)

显卡内存不足

*** Check failure stack trace: ***

Process finished with exit code -1073740791 (0xC0000409)

问题描述:

训练AI模型时,遇到无错跳出finished with exit code -1073740791 提示,显卡内存不足。

原因分析:

很有可能是你只有单个显卡,一次读取的数据太大,从而内存溢出;


解决方案:

解决方式一:训练时可以适当的调小batch_size;
方法二,换显卡或到多显卡的机器上去训练。

猜你喜欢

转载自blog.csdn.net/beauthy/article/details/108323989
今日推荐