YOLO深度学习(计算机视觉)—毕设笔记(yolo训练效率加快)

今天在网上看到一篇教学分享,一个博主说很多人有个错误的观念:虽然可以使用GPU跑模型,但是大家都认为自己电脑只有一个显卡,然后参数workers>0的时候就会报错,所以默认都设置workers=0就不会报错。

但实际这样并没有最大程度利用我们的GPU,我们输入命令【nvidia-smi】可以看到我们的显卡信息就可以看到我们是并没有充分利用的

解决方法就是:

  • 把训练代码放到 if __name__ == '__main__': 代码块里
    • Windows 下开启 workers>0 必须满足用【if __name__ == '__main__':】包裹训练逻辑
  • 在块的最开头加 torch.multiprocessing.freeze_support()

例子

速度将会直线飞升!!!!!!!!

相关推荐
小徐xxx20 小时前
Softmax回归(分类问题)学习记录
深度学习·分类·回归·softmax·学习记录
藦卡机器人20 小时前
国产分拣机器人品牌有哪一些做的比较好的推荐?
人工智能
GJGCY20 小时前
2026主流智能体平台技术路线差异,各大平台稳定性与集成能力对比
人工智能·经验分享·ai·智能体
AAD5558889920 小时前
YOLOv8-MAN-Faster电容器缺陷检测:七类组件识别与分类系统
yolo·分类·数据挖掘
橙露20 小时前
视觉检测中的数字光纤放大器的核心参数和调整
人工智能·计算机视觉·视觉检测
Rorsion20 小时前
机器学习过程(从机器学习到深度学习)
人工智能·深度学习·机器学习
JicasdC123asd20 小时前
【工业检测】基于YOLO13-C3k2-EIEM的铸造缺陷检测与分类系统_1
人工智能·算法·分类
咚咚王者20 小时前
人工智能之核心技术 深度学习 第十章 模型部署基础
人工智能·深度学习
ydl112820 小时前
深度学习优化器详解:指数加权平均EWA、动量梯度下降Momentum、均方根传递RMSprop、Adam 从原理到实操
人工智能·深度学习
幂链iPaaS20 小时前
市场六大专业iPaaS平台怎么选
大数据·人工智能