yolov5gpu训练出来的模型无效

刚开始我是用 gpu 去训练的模型,后来发现训练出来的模型识别不出来物体最后仔细观察了一下发现红框里面的参数是 nan。
然后用 cpu 去训练了一下红框里就有参数了,
有人知道为什么我用
gpu去跑这些参数是nan 吗?

img

不知道你这个问题是否已经解决, 如果还没有解决的话:
  • 关于该问题,我找了一篇非常好的博客,你可以看看是否有帮助,链接:在云端GPU训练yolov5详细步骤
  • 除此之外, 这篇博客: Ubuntu yolov3 Gpu 安装及训练个人数据(奶妈教程)中的 产生问题原因: 显卡版本过高 降低版本:先卸载后安装低版本 操作如下 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
    sudo apt-get remove --purge nvidia-* (这个是使用自动安装的卸载方式)
    sudo ./NVIDIA-Linux-x86_64---xxx.run  --uninstall (这个是使用.run 文件安装的卸载方式,xxx 代表版本号)
    !开始安装
    sudo service lightdm stop (关闭图形界面)
    sudo add-apt-repository ppa:graphics-drivers/ppa
    sudo apt-get update
    sudo apt-get install nvidia-driver-460
    reboot
    进入图形界面
    nvidia-smi
    查看是否安装成功
    

    在这里插入图片描述
    这个是这个

    老哥那拿来的,因为我不想马克:
    可以看到左上角 NVIDIA-SMI是他的驱动型号 ,右上角是推荐安装的CUDA的型号。即使你没安装也会有。这个推荐是最高版本。向下兼容,推荐安装10.2的版本。很稳定!!!!!!

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^