用Yolo5跑自己的数据集,torch_GPU和CUDA都安装了,使用GPU跑模型,GPU的占用率为1%,设置的epoch为300,batch size为32,大概6min多/epoch,数据集很小,刚过一千。在网上查了相关的文章,发现可能是读取数据造成的,把cache-image的action='store_true'改成'store_false'。
改成
修改后 每epoch大概是13s!超级快,GPU的占用率变到33%,CPU的占用率50%。文章来源:https://www.toymoban.com/news/detail-534823.html
后面再想办法改一改,计算机小白学习好难,论文好难(哀嚎~)!文章来源地址https://www.toymoban.com/news/detail-534823.html
到了这里,关于YOLOv5训练速度慢 GPU占用率低的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!