没有这样的操作员 torchvision::nms

Question

当我尝试运行 yoloV3 检测时，发生了错误

op = torch._C._jit_get_operation(qualified_op_name)

RuntimeError: No such operator torchvision::nms

虽然这段代码是 torchvision 的源代码，但我尝试了一些时间来通过失败的提示来纠正代码。

Answer 1

由于我在安装 torch 和 torchvison 时遇到了很多障碍，我不愿意重新安装环境。运行“conda list”torch 和 torchvison 的版本，我发现它们并不是不兼容的，安装的两个软件包的版本是：

\n\n

 torchvision-0.5.0+cu92-cp37-cp37m-win_amd64.whl\n torch-1.4.0+cpu-cp37-cp37m-win_amd64.whl\n

\n\n

我将torchvison的版本更改为

\n\n

 torchvision-0.5.0+cpu-cp37-cp37m-win_amd64.whl\n

\n\n

到目前为止我已经处理了我的问题。\nnote\xef\xbc\x9a 我的操作系统和 python 的环境是 Fellowing\xef\xbc\x9a

\n\n

  Windows_X64\n  python3.7\n

\n