纯干货!!!从零开始训练模型
一、环境的安装
1.下载安装GIT
###如果已经安装GIT则请跳过
###本文档所有有下划线的都可以Ctrl+鼠标左键直达连接,如果加载不出来请使用科学多试几次。
###下载这个软件是一方面是为了下载ultralytics## 标题包,下载这个包有很多种路径,可以通过官网直接下载Zip文件,官网如下GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite(下载完后会得到一个ultralytics—main文件,可以将文件名改为ultralytics即可),另一方面因为后续需要扒其他的包,为了方便学习这里推荐下载。
1.1下载Git
下载:Git - 下载软件包 (git-scm.com)
安装教程参考:跟着走,很快能装完Git 详细安装教程(详解 Git 安装过程的每一个步骤)-CSDN博客
2.部署YOLOV8项目
###自己选择自己将要训练模型的路径,建议不要放到系统盘,因为以后会训练大量的数据集,很占内存!!!
2.1选择路径
###选好路径之后,右键点击Open Git Bash here,进入Git窗口。
2.2扒取项目
###这里如果一次下载不好切换加速器,多试几次
git clone https://github.com/ultralytics/ultralytics.git
2.3扒取结束
2.4下载权重
###进入下载好的ultralytics文件中,点击链接进入官方文档:YOLOv8 - Ultralytics YOLOv8 文档,点击YOLOv8n进行下载,把下载好的文件放入上面的ultralytics文件中。
3.安装CUDA
###这里一定要根据自己支持的版本下载。
3.1查看显卡
###win+R打开命令提示符窗口,输入以下命令,查看自己的显卡最高支持CUDA的哪个版本,这里可以看到我的最高可以支持12.0版本。
nvidia-smi
###为了检测显卡驱动的问题,如果显示不出来则需要自己装显卡驱动
3.2检查CUDA的安装
###如果未安装,请跳过下一步直接看3.4
nvcc -V
3.3CUDA的卸载
###若安装了建议还是卸载在跟着我安装一次,因为这里面有坑,跟PYtorch,cudnn都有关。卸载如下:
3.3.1卸载CUDA
###搜索框搜索应用和功能进入界面,搜索NVIDIA,卸载我标注的五个带版本号的(就是标了CUDA和版本号的),不要乱卸载,注意看清楚,一定要卸载带版本号的!!!别问我问什么QAQ
3.3.2删除环境变量
###打开将高级系统设置里的环境变量,找到PATH,点击编辑,删除这四个带有CUDA的环境变量
3.4CUDA的安装
3.4.1CUDA的版本查看
###因为这里CUDA是为了配合Pytorch使用,所以先去pytorch官网查看CUDA版本(PyTorch),如果CUDA不是11.8和12.1的,在历史版本里找到自己能用的版本,查看自己下载哪个版本的CUDA