yolov5用官方权重文件测试detect.py检测无结果
运行过程中没有报错,但没有任何结果。
yolov5用官方权重文件测试detect.py检测无结果
原因很简单,就是你的cuda版本错误了。英伟达的显卡从30系开始,就不支持cuda11.0以下的cuda版本了,你这个40系的需要安装cuda11以上的cuda版本才行。
所以解决方案就是换cpu,或者重新安装cuda,然后torch等和cuda相关的都要换成cuda11.x的版本。一般来说,cuda11.x只要大于torch要求的11.y(x>=y)是没啥问题的,所以你可以安装cuda11.8,然后torch和torchvision等安装就看你要多少版本了。另外,请注意,cuda目前最新的是12.x的版本,目前阶段不是很建议用12.x版本的cuda