ValueError: Could not find the operator torchvision::nms.

Question

ValueError: Could not find the operator torchvision::nms. Please make sure you have already registered the operator and (if registered from C++) loaded it via torch.ops.load_library.请问这个问题要怎么解决

Answer 1

参考通义千问和郭老师的小迷弟雅思莫了-编写提供，主要是为了帮助您快速解决问题，如下回答，您可以参考下，如有问题，评论区留言。

这个错误信息表示在运行过程中，PyTorch无法找到torchvision::nms（非极大值抑制）这个操作。非极大值抑制通常用于目标检测任务中，它是torchvision库的一部分。

解决这个问题的步骤如下：

1. 确保你已经正确安装了torchvision库，并且其版本与你的torch库兼容。你可以通过以下命令进行安装或更新：

pip install torchvision -U

1. 如果你是在自定义C++扩展中使用了该操作符，并且编译了自己的库，那么你需要确保在Python代码中正确加载了这个库。例如：

import torch
# 假设你的C++扩展库名为'custom_ops.so'
torch.ops.load_library('custom_ops.so')

1. 对于大部分情况，直接使用torchvision中的nms函数无需上述额外步骤。如果你在标准用法下遇到此问题，可能需要检查torchvision的版本或者是否存在环境配置问题。

2. 如果以上方法都无法解决问题，尝试重新创建一个新的虚拟环境，并在此环境中仅安装torch和torchvision最新版，看是否能正常工作。