问题:
今天想玩下yolov5,从官网把yolov5的代码下载下来之后,配置了GTX1650的pytorch GPU版本。
一开始是用cpu跑得测试图,发现可以检测出来(有检测框)。
后来想用GPU版本的试一下,在一番操作解决了各种库的版本问题后,发现GPU的确在训练模型。训练完成之后想用GPU检测一下看看效果,发现却没有检测框。
原因:
GPU没有达到CUDNN_HALF使用要求!!!
解决方案:
在detect.py中添加以下代码:
torch.backends.cudnn.enabled = False
成功!!!