weixin_64202324的博客

原创 16系显卡在YOLOv10训练时检测出现nan值的解决方法

这个问题我在用yolov5时其实就已经遇到了，当时在网上到处查教程，最后的解决方案是用cuda10.2的版本，虽然速度慢了一些，但是好歹能跑起来了。这次yolov10又出这个岔子，可是yolov10要求的torch版本对应的cuda版本最低都是11.7版本了，降级这一招行不通了（醉了）。我就又去网上查相关解决方法，不过都没有yolov10这个版本的对应的解决方法。按网上的教程把环境配置好后，跑起来却出现了训练时检测出现nan值的情况（如下图），很生气。