YOLOv5模型训练避坑指南:从环境配置到loss震荡的实战解析
刚接触YOLOv5时,我被它简洁高效的特性吸引,但很快发现从环境配置到模型训练的路上布满"暗坑"。尤其在Windows系统下,PyCharm环境的各种兼容性问题让新手举步维艰。本文将分享我在YOLOv5实战中积累的排雷经验,帮你避开那些浪费时间的陷阱。
1. 环境配置:从零搭建稳定训练基础
1.1 Python环境与PyCharm配置要点
在Windows上使用PyCharm开发YOLOv5时,Python版本选择至关重要。官方推荐Python 3.8-3.10,但实测发现:
- Python 3.9与最新版PyTorch组合最稳定
- 避免使用conda默认创建的base环境,建议新建专用虚拟环境
bash复制# 创建虚拟环境(PyCharm终端执行)
python -m venv yolov5_env
.\yolov5_env\Scripts\activate
常见报错:若出现无法加载文件...因为在此系统上禁止运行脚本错误,需以管理员身份运行PowerShell执行:
powershell复制Set-ExecutionPolicy RemoteSigned
1.2 PyTorch与CUDA的版本匹配陷阱
YOLOv5对PyTorch版本敏感,版本不匹配会导致各种诡异错误。根据显卡型号选择正确组合:
| 显卡架构 | 推荐PyTorch版本 | CUDA版本 | 备注 |
|---|---|---|---|
| NVIDIA 30系 | 1.12.1+cu116 | 11.6 | 需手动安装 |
| NVIDIA 20系 | 1.10.2+cu113 | 11.3 | 较稳定 |
| 笔记本显卡 | 1.8.0+cu111 | 11.1 | 兼容性好 |
安装命令示例(根据实际CUDA版本调整):
bash复制pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/torch_stable.html
提示:
