pytorch - 没有这样的运营商 torchvision::nms

Question

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

op = torch._C._jit_get_operation(qualified_op_name)

RuntimeError: No such operator torchvision::nms

虽然这段代码是torchvision的源代码，但我尝试了几次通过失败的提示来更正代码。

标签： pytorchtorchvision

Answer 1

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

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

我将 torchvison 的版本更改为

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

到目前为止，我已经处理了我的问题。注：我的os和python的环境正在交友：</p>

  Windows_X64
  python3.7