项目场景:
操作系统:windows10
硬件设备:RTX1650(其他可能出现该问题的显卡:游戏本显卡1650,1660以及1660Ti)
问题描述
yolo系列运行成功后,在cpu中能跑出检测框来,gpu中识别不出检测框。
c语言代码:
darknet_no_gpu.exe detect yolov3.cfg yolov3.weights dog.jpg
cpu用时:1148.185000 milli-seconds.
darknet.exe detect yolov3.cfg yolov3.weights dog.jpg
gpu用时:517.685000 milli-seconds.
python语言代码:
python detect.py --weights yolov7.pt --source bus.jpg --no-trace --device cpu
cpu用时:(720.1ms) Inference
python detect.py --weights yolov7.pt --source bus.jpg --no-trace
gpu用时:(231.3ms) Inference
原因分析:
猜测原因:RTX1650没有达到CUDNN_HALF使用要求
解决方案:
总结:需要关闭CUDNN_HALF
一、yolov3;Visual Studio 2019;c语言
方法一:右键属性 -> C/C++ -> 预处理器定义 -> 删除cudnn_half
方法二:使用记事本打开darknet.vcxproj -> 编辑 -> 查找 ->删除cudnn_half
二、yolov7;PyCharm Community Edition 2021;python语言
在主函数中添加:
torch.backends.cudnn.enabled = False