RuntimeError: No such operator torchvision::nms

今天遇到了如题目所示的问题

项目地址:naver/dope

由于是个新的项目,还没有issues,只能自力更生。

查了很多博客和其他项目的issues。

基本上都说的是torch和torchvision的版本不匹配造成的。

看到的是如下的截图:

但是没有解决问题,毕竟本地的版本是匹配的。

后来在这个issue里得到了灵感。

既然pythorch作者会修改源码来修复一些已知的问题,那么会不会错误的原因是pytorch版本问题呢?

于是开始求证。

在项目中,作者说用的是pytorch的1.5版本和torchvision的0.6版本。

所以在安装环境的时候,torch==1.5默认安装1.5.0版本。

再回到torchvision的网站,给出的版本对应信息如下:

果然1.5版本有个1.5.1,把torch和vision都换成相应版本,问题解决。

再列两个虽然没解决问题,但个人认为很有帮助的issues:

issue1

issue2