YOLO系列训练时出现loss出现nan值或者测试时P\R\map全部为0值

GTX16xx系列显卡，显卡为GTX 16XX系列的在cuda使用较新版本时会出现该问题。

解决方案 

安装cuda，cuda换为10.2的版本，注意如果已经安装cuda需要选自定义安装，只选择第一项cuda

https://developer.nvidia.com/cuda-toolkit-archive

cudnn下载:

https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b

 解压下再好的cudnn

把这三个文件夹 ，里的内容分别放在

cuda的

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64

然后继续安装pytorch cu102版本

 pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

再次训练，问题解决