跑yolov5遇到的问题

问题一

跑yolov5遇到的问题_第1张图片
选好模型和资源路径后开始跑,报错:
No such operator torchvision::nms
这是没装torchvision或者是版本太低
pip install torchvision 自动下载了0.10.0版本

问题二

上面这个问题解决了又出现一个新的问题:
Couldn’t load custom C++ ops.
这是由于torchvision与torch版本不匹配的问题,我调用torchvision.__version__发现版本是0.10.0但是我的torch是1.9+cu102所以我要下载torchvision 0.10.0+cu102版本的才行。
进入https://download.pytorch.org/whl/torch_stable.html
找到
跑yolov5遇到的问题_第2张图片
注意要下载上面那种的,就是有%2Bcu102的(代表+cu102),下面的还是cpu版本的,其他参数在上一篇博文说过了。
输入指令:

pip install https://download.pytorch.org/whl/cu102/torchvision-0.10.0%2Bcu102-cp39-cp39-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

但是在我这个时候,清华镜像网站上没有这个东西,然后我又没有tizi,导致速度太慢,下到一半直接卡住了,超时报错。
另一边,我直接点击蓝字下载.whl文件,试了5次直接秒下。
在这里插入图片描述
右键点击属性获得路径
用命令在本地安装:

pip install D:\360极速浏览器下载\torchvision-0.10.0+cu102-cp39-cp39-win_amd64.whl

搞定。

你可能感兴趣的:(安装)