基于Windows系统进行Yolov3训练模型遇到:“CUDA Error: out of memory”的解决方案

时间:2024-04-03 13:59:54

基于Windows系统进行Yolov3训练模型遇到:“CUDA Error: out of memory”的解决方案

 

遇到问题:CUDA Error: out of memory

基于Windows系统进行Yolov3训练模型遇到:“CUDA Error: out of memory”的解决方案

 

原因:爆显存  Hhhhhha !!!

 

解决方案:更改训练所使用的CFG(.cfg)文件参数  (本人使用的是:yolov3-voc.cfg)

(1)修改  batch subdivisions 值   (batch可较默认值64略小,subdivisions可

较默认值48略大),此处设置:batch=48, subdivisions=48

基于Windows系统进行Yolov3训练模型遇到:“CUDA Error: out of memory”的解决方案

 

(2)修改尾行 random 值  (默认值为1),此处修改为 :random=0

基于Windows系统进行Yolov3训练模型遇到:“CUDA Error: out of memory”的解决方案

 

 

成功解决:

基于Windows系统进行Yolov3训练模型遇到:“CUDA Error: out of memory”的解决方案

 

如若还不行,基于这三个参数略微变动,即有大概率可解决。