YOLOV5训练数据集P、R、mAP等均为0的解决方案

使用YOLOV5训练数据集时，P、R等值均为0

最近在用YOLOV5训练自己的数据集，执行了十几个epochs之后，epoch的主要参数，比如box，obj，cls，labels等均为nan或0，即没有跑通正确的结果，找了很多方案，最终发现是pytorch版本不适合。

我的cuda版本是10.2，pytorch版本是1.7.1，重新安装了1.8.0版本的（pytorch-1.8.0-py3.8_cuda11.1_cudnn8_0.tar.bz2）以及相应的torch1.8.0（cu102/torch-1.8.0-cp38-cp38-win_amd64.whl）以及与之匹配的torchvision（torchvision-0.9.0-py38_cu102.tar.bz2）后，成功解决了这一问题。
torch和torchvison的匹配可以参考如下链接：https://codechina.csdn.net/mirrors/pytorch/vision?utm_source=csdn_github_accelerator

附pytorch下载链接https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

torch和torchvision下载链接https://download.pytorch.org/whl/torch_stable.html

yolov5训练自己的数据集过程中还遇到一些其他问题，在我的另一篇博客有详细说明https://blog.csdn.net/weixin_42475780/article/details/121078534