YOLOv5s GTX 1660 Ti训练时出现,box,obj,cla全是nan的问题P、R、mAP都是0,Pytorch和cuda、cudnn版本不对

这里写自定义目录标题

  • 参考资料
  • 问题描述
    • 当前使用版本
    • 尝试解决问题版本(1)
    • 尝试解决问题版本(2)
    • CUDA卸载参考
    • CUDA卸载之后发现依然还在,需要把torch卸载掉。
  • 1660Ti出现的nans问题以及PR为0问题

参考资料

1.Github_YOLOv5_nan问题

2.查看CUDA最高可使用版本_nvidia-smi

3.CSDN问题描述参考链接

4.1 Pytorch与cuda、cudnn版本对应关系1

4.2 cuda与torch版本对应关系2
(感觉2比1 正确)

5.torch、cuda版本源

问题描述

YOLOv5s GTX 1660 Ti训练时出现,box,obj,cla全是nan的问题P、R、mAP都是0,Pytorch和cuda、cudnn版本不对_第1张图片
训练显示box、obj、cls都是nan

当前使用版本

pytorch版本:1.9.0
cuda版本:11.1
cudnn版本:8.0.5

尝试解决问题版本(1)

Pytorch版本:1.9.0
cuda版本:11.4
cudnn版本:8.2.2(或者8.2.4)

尝试解决问题版本(2)

Pytorch版本:1.9.0
cuda版本:10.2
cudnn版本:8.2.2

CUDA卸载参考

参考链接
控制面板-程序-卸载:CUDA的全部卸载掉。

CUDA卸载之后发现依然还在,需要把torch卸载掉。

pytorch和torchvision的安装卸载指南点这里

1660Ti出现的nans问题以及PR为0问题

我的环境是:
torch==1.13.1+cu116
torchvision==0.14.1+cu116
torchaudio==0.13.1
cudnn==8304
出现了nan和P、R、mAP都是0的情况

你可能感兴趣的:(环境问题,pytorch,YOLO,深度学习)