PIP安装YOLOv5

首先需要创建conda环境：

conda create -n y5pip python=3.8

创建完成后，进入环境

conda activate yolov5pip

pip安装yolov5，注意这里安装的yolov5版本和你使用的模型版本要一致，如果使用官方的yolov5s这些模型，要使用5.0以上版本，如果使用自己训练的模型，要下载与训练模型相同的版本

pip install yolov5==7.0（这里填版本号）

安装ultralytics

pip install ultralytics==8.0

到这里就可以使用cpu进行训练检测了

yolov5 detect --source xxx --weight xxx 
	   train  --data xxx --weight xxx

上面的参数可以参考完整版的yolov5里的配置项，把需要修改的配置项按上面的写法写上

配置gpu训练，需要先卸载原有的torch版本

pip uninstall torch
pip uninstall torchvision
pip uninstall torchaudio

安装下述版本

pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0+cu118 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package --trusted-host mirrors.aliyun.com

根据自己的cuda版本就行更改torch==2.1.0+cu118，就是安装2.1版本torch，cuda版本11.8，查看自己版本可以打开命令行输入nvidia-smi，也可以进入torch官网选配pytorch官网
安装完成后再进行上述测试