问题描述:
运行Yolov5测试文件报错,该报错说明torch版本不适配。
解决办法:
在Anaconda prompt打开该虚拟环境–>conda list,发现torch版本如下:
torch 1.8.0
torchvision 0.9+cu111
我们需要的是“torch 1.8.0+cu111”这说明我们的torch版本错误,采用以下两个命令之一去卸载torch:
conda uninstall torch
pip uninstall torch
采用以下命令下载新的torch:
pip install "torch-1.8.0+cu111-cp37-cp37m-win_amd64.whl" -i http://pypi.douban.com/simple/
再次运行Yolov5-master下面的detect.py文件结果如下:
通过提示我们找到“run\detect\exp4”下的图,这下就运行成功了