问题描述:

运行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”下的图,这下就运行成功了

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐