RuntimeError: No such operator torchvision::nms的解决方法
1.RuntimeError: No such operator torchvision::nms的解决方法。
在使用github上的yolov4源码时遇到这个问题。查阅资料发现可以通过调整python,pytorch,torchvision版本解决这个问题。
本人出现这个报错的版本为python3.6,pytorch1.6,torchvision0.7.0.
重新创建新环境成python3.7,pytorch1.7,torchvision0.8.1.成功解决该问题。值得注意的是有同样出现此问题的人员反映重新创建环境版本为python3.8并没能解决该问题。具体方法如下:
新建一个虚拟环境并安装3.7的python环境
在Windows环境下的anaconda中指令为conda create -n your_env_name python=3.7
。命令创建python版本为3.7、名字为your_env_name的虚拟环境。your_env_name文件可以在Anaconda安装目录env