YoloV5环境搭建记录

https://github.com/ultralytics/yolov5/

1、在Anaconda Promptx新建conda虚拟环境

conda create -n py39_yolov5 python=3.9

2、激活虚拟环境

conda activate py39_yolov5

3、虚拟环境下装所需依赖

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cpuonly -c pytorch  ---------这里使用cpu

# CUDA 11.6
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1
cudatoolkit=11.6 -c pytorch-------有英伟达显卡的话可以用这个

4、下载yolov5和权重文件，并将权重文件放入yolov5下载的目录里

进入到yolov5目录并安装依赖pip install -r requirements.txt
安装时遇到编码问题

修改编码也没用，后面通过升级conda（ conda update -n base conda）后重新安装pip install解决

安装过程因网络问题较慢，需等待较久

5、验证测试yolov5环境是否安装ok

在runs\detect\exp目录下有检测的结果图片

6、测试下训练

python train.py

训练中出错

按如下修改后重新train

cpu训练很慢，pc会很卡

 训练结果会在runs\train\exp生成