1. 问题
40系列显卡训练yolo系列出现问题,loss正常,但mAP,R,P等为零。
环境:ultralytics版本为8.3.9,cuda11.8, torch2.4。
40系列显卡网上说可以使用cuda低于11.7的,自己测试了下11.3和touch1.12.0 还是不行,torch.cuda.is_avaliable()依然为False。
无法像网上说的map改为False, half=False,仍然没用。
2. 解决办法
自己尝试将ultra版本改为8.2.9,就正常了,但是这个use使用不了yolov11最新,最多10了。先用着试一试,后面再看升级cuda试一试。遇同样问题的可以评论区交流交流。