关于我初次尝试使用yolov5进行模型训练的成功的总结

用yolov5-6.0训练出一个能用来识别街头小广告的模型（也可以被叫做权重，类似yolov5s）。
下面是我这几天来，看了无数个教程得到的总思路，这其中走过的坑实在太多了，解决的bug也太多了，但是当你看到模型训练出来的时候，你会觉得这所有的苦都是值得的。

第一步：重装anaconda

为什么我把这一步叫做重装anaconda呢？是因为我的anaconda是在学习大学课程python的时候安装的老师的安装包。在我进行了很多步骤以后，发现打不开Anaconda Navigator和spyder了，如下图

我从网上找了很多的教程，包括
下图很经典的方法

但是执行到第三步就开始报错了，我在网上找了很多的方法，都说是pyqt5的问题，可是最后也没有找到和我情况一致的。所以我决定，直接重装anaconda，先卸载干净再安装新的，关于怎么卸载干净可以从网上找一下，我这里有一个可以参考的链接
如何彻底卸载Anaconda？
根据他的方法可以卸载，完成后会存留一个备份，里面装着一些虚拟环境文件等等。这样比较安全，最后重装，重装的话直接到清华源里下载，比较快，点这里。我装的是2020年版本的anaconda，基础python版本是3.8，这也正好符合yolov5需要的python版本，（yolov5需要python版本>3.7）比较稳定。

当我重装完成以后，终于可以打开Anaconda Navigator和spyder了（欣喜若狂）。所以总结一下，anaconda遇到实在解决不了的问题，就重装，我在这个环节整整卡了两个晚上。
安装完成后，要新建一个自己的虚拟环境，下面所有的操作都要在这个虚拟环境中进行，我创建的环境名称是yolo。

第二步：安装cuda和pytorch

安装cuda是分两步的，一步是安装cuda，一步是下载cudnn。这个我就直接放链接了，CUDA和CUDNN的安装（带有详细步骤）
我没啥好总结的，因为好早就装完了，注意的是这个针对的是想用自己的显卡进行模型训练的，不是显卡的这不用这一步。

然后就是pytorch了，这个又是个卡了我老半天的步骤。在安装pytorch的时候，我遇到了几个难题。
1.我想在清华源下载，因为快，但是不知道该下哪几个压缩包。
2.下完了压缩包怎么安装。
3.安装完了为什么import torch还是报错（困扰我最长时间的）。
针对这三个问题，我依次给出解决方法。
第一个问题需要安装相应版本的torch、torchaudio和torchvision。注意要与你的python版本和安装的cuda版本
相对应。
第二个问题需要一个命令：
conda install --offline 加上压缩包的路径
例如：

conda install --offline C:\ProgramData\Anaconda3\pkgs\pytorch-1.11.0-py3.8_cuda11.5_cudnn8_0.tar.bz2

conda install --offline C:\ProgramData\Anaconda3\pkgs\torchaudio-0.11.0-py38_cu115.tar.bz2

conda install --offline C:\ProgramData\Anaconda3\pkgs\torchvision-0.12.0-py38_cu115.tar.bz2

关于第三个问题，为什么完成上面两个步骤以后import torch 还是报错呢？我搜集了全网很多的资料，明白了安装pytorch不止有pytorch、torchvision和torchauduio这三个包，还有其他的一些依赖，所以最后找到一个最好的办法。
首先，在pytorch官网找到适合自己情况的conda代码，然后

在自己的yolo环境中执行这行代码时注意要删掉-c pytorch，因为这样才能在清华源安装，否则就是在pytorch网安装。（前提是自己的安装源已经是清华源的了，这一步网上都有，channels代码我放下面了，把它复制到.condarc里面去）

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

重点来了，只执行上面的命令是不够的，因为使用此命令在清华源里面默认是下载cpu版本的torch和torchvision和torchaudio。所以我们要在执行完成后，再把执行上面第二个问题的步骤。

conda install --offline C:\ProgramData\Anaconda3\pkgs\pytorch-1.11.0-py3.8_cuda11.5_cudnn8_0.tar.bz2

conda install --offline C:\ProgramData\Anaconda3\pkgs\torchaudio-0.11.0-py38_cu115.tar.bz2

conda install --offline C:\ProgramData\Anaconda3\pkgs\torchvision-0.12.0-py38_cu115.tar.bz2

这样最终是安装完全了，这些是我跟着一个博主的文章学的，这个文章很清楚地指出了问题所在。清华源conda 安装gpu版本的pytorch总是推送cpu版本解决办法

干完这些，import torch就没有问题了。
然后再跟着网上的其他教程走就可以。
第三步：进行自己的项目
这一步我就不多赘述了，参考这篇教程就可以。
[YOLOv5]如何训练自己的数据集（GPU版）
注意在配置requirements.txt环境的时候要把里面的torch和torchvision注释掉，因为前面已经安装了gpu版。