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

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

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

解决方法就是:

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

例子

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

相关推荐
Nautiluss2 小时前
一起调试XVF3800麦克风阵列(十六)
人工智能·单片机·音频·语音识别·dsp开发·智能硬件
AI街潜水的八角2 小时前
基于深度学习神经网络YOLOv4目标检测的汽车车牌识别系统
深度学习·神经网络·yolo
人工智能AI技术2 小时前
安卓AI智能体开发实战:基于AndroidGen-GLM搭建手机端超级助理,支持离线运行
人工智能
Hello_Embed2 小时前
串口面向对象封装实例
笔记·stm32·单片机·学习·操作系统
杨浦老苏2 小时前
轻量级自托管笔记与清单利器jotty·page
笔记·docker·markdown·todo·群晖
指掀涛澜天下惊2 小时前
概率论 - 贝叶斯定理
人工智能·机器学习·概率论·贝叶斯定理·贝叶斯公式
CCPC不拿奖不改名2 小时前
“Token→整数索引” 的完整实现步骤
人工智能·python·rnn·神经网络·自然语言处理·token·josn
deephub2 小时前
多智能体强化学习(MARL)核心概念与算法概览
人工智能·机器学习·强化学习·多智能体
三伏5222 小时前
stm32f103系列手册IIC笔记
笔记·stm32·嵌入式硬件