之前折腾了yolov3
现在直接上yolov5，话不多说。
（1）安装cuda
首先打开官网
https://pytorch.org/get-started/locally/
我们可以看到需要的是cuda11
直接
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge查看一下版本直接yes
安装完成
（2）测试
发现问题
解决：版本问题
不得不说还是这个老哥的版本强呀！
https://www.cnblogs.com/winslam/p/13474330.html
输入下面的代码
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.htmlpip install ipython就能把gpu加速的版本下下来了
测试一下，测试代码
python
import torchtorch.cuda.is_available()
import torchvision
torchvision.__version__结果
（3）下载代码
进入官网下载代码
https://github.com/ultralytics/yolov5
直接进入文件夹
输入代码
pip install -r requirements.txt
完成
（4）测试
我现在主要是在pychram环境里面设置的
关于pychram怎么导入anaconda环境并且配置看这篇文章，里面有详细的介绍
https://blog.csdn.net/qwazp3526cn/article/details/115299589
在pychram环境下面的终端输入
python detect.py --source ./data/images/90.png可以看到结果
我们打开runs\detect\exp3文件夹
可以看到自己的数据集跑出来的结果
