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

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

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

解决方法就是:

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

例子

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

相关推荐
pp起床3 分钟前
Gen_AI 第四课 模型评估
人工智能
zhangshuang-peta5 分钟前
人工智能代理团队在软件开发中的协同机制
人工智能·ai agent·mcp·peta
love you joyfully5 分钟前
告别“人多力量大”误区:看AI团队如何通过奖励设计实现协作韧性
人工智能·深度学习·神经网络·多智能体
2501_945318498 分钟前
AI证书避雷,需认准官方资质与行业口碑两大核心
人工智能
方见华Richard8 分钟前
世毫九“量子原住民”教育理念完整框架
人工智能·交互·学习方法·原型模式·空间计算
一切尽在,你来9 分钟前
1.3 环境搭建
人工智能·ai·langchain·ai编程
njsgcs9 分钟前
agentscope 调用vlm
人工智能
happyprince13 分钟前
2026年02月08日热门论文
人工智能·深度学习·计算机视觉
七牛云行业应用14 分钟前
1M上下文腐烂?实测Opus 4.6 vs GPT-5.3及MoA降本架构源码
人工智能·python·llm·架构设计·gpt-5·claude-opus
芷栀夏17 分钟前
CANN ops-math:面向 AI 计算的基础数学算子开发与高性能调用实战指南
人工智能·深度学习·神经网络·cann