【pytorch使用报错】No such operator torchvision::nms

项目场景：

由于项目使用的是低版本pytorch训练的，搭建该项目yolov5环境时遇到该报错

---

问题描述

---

原因分析：

在网上搜索有人说由于torch版本和torchvision的版本不匹配造成的。
但torch1.4.0与torchvision0.5.0的相配的。后来突然想起来，我torch1.4.0是使用conda安装的，而torchvision是直接在官网地址下载whl文件安装的。

---

解决方案：

统一两个安装包，要不然就都使用conda来安装cuda10.0版本的torch和torchvision，要不然就都在官网下载whl来安装。

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.0 -c pytorch

不过不知道清华源是不是删改了一部分版本，我输入上述命令，提示找不到torchvision的这个版本。所以我才去官网下载whl安装的，这也就导致了上面的报错。
官网地址如下：
torch官网whl下载地址

将红框里的两个文件下载到本地安装。命令如下，可以根据自己实际下载文件名修改

python -m pip install torch-1.4.0+cu100-cp37-cp37m-linux_x86_64.whl

python -m pip install torchvision-0.5.0+cu100-cp37-cp37m-linux_x86_64.whl

这样再运行程序。测试正常了。