RuntimeError: No such operator torchvision::nms的解决方法

最新推荐文章于 2024-10-25 09:57:10 发布

牧濑_红莉栖

最新推荐文章于 2024-10-25 09:57:10 发布

阅读量6.1k

点赞数 2

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/weixin_43843824/article/details/109448314

在尝试运行github上的yolov4源码时遇到RuntimeError: No such operator torchvision::nms的问题。通过创建新的虚拟环境，安装python 3.7，pytorch 1.7和torchvision 0.8.1，该问题得以解决。创建虚拟环境的步骤包括在anaconda中指定python 3.7版本，激活新环境，并根据PyTorch官网的指示安装对应版本的PyTorch。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

RuntimeError: No such operator torchvision::nms的解决方法

1.RuntimeError: No such operator torchvision::nms的解决方法。
- 新建一个虚拟环境并安装3.7的python环境

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安装目录envs文件下找到。
用activate your_env_name进入新建的虚拟环境。
在https://pytorch.org/中选择对应版本要求。
在这里插入图片描述
复制最下面的命令语句在新建的环境中执行。
新环境安装完成后采用新环境中的解释器即可解决问题。