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

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

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

解决方法就是:

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

例子

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

相关推荐
CV-杨帆2 分钟前
ICLR 2026 LLM安全相关论文整理
人工智能·深度学习·安全
智者知已应修善业8 分钟前
【51单片机单按键切换广告屏】2023-5-17
c++·经验分享·笔记·算法·51单片机
田八9 分钟前
聊聊AI的发展史,AI的爆发并不是偶然
前端·人工智能·程序员
zandy101110 分钟前
全链路可控+极致性能,衡石HENGSHI CLI重新定义企业级BI工具的AI协作能力
大数据·人工智能·ai analytics·ai native·agent-first
广州灵眸科技有限公司11 分钟前
为RK3588注入澎湃算力:RK1820 AI加速卡完整适配与评测指南
linux·网络·人工智能·物联网·算法
小程故事多_8012 分钟前
从零吃透Transformer核心,多头注意力、残差连接与前馈网络(大白话完整版)
人工智能·深度学习·架构·aigc·transformer
xiejava101814 分钟前
写了一个WebDAV的Skill解决OpenClaw AI助手跨平台协作难题
人工智能·ai编程·智能体·openclaw
zhanghongbin0120 分钟前
AI 采集器:Claude Code、OpenAI、LiteLLM 监控
java·前端·人工智能
AI应用实战 | RE23 分钟前
012、检索器(Retrievers)核心:从向量库中智能查找信息
人工智能·算法·机器学习·langchain
IT_陈寒25 分钟前
Python的列表推导式里藏了个坑,差点让我加班到凌晨
前端·人工智能·后端