RuntimeError: No such operator torchvision::nms问题解决

作者今天在跑之前写过的一个算法时遇到了这个问题。torch和torchvision版本和之前用的时侯没有变化,torch==1.4.0+cu100 torchvision==0.5.0+cu100。并且

 import torch
 torch.ops.torchvision.nms

输出了

RuntimeError: No such operator torchvision::nms

问题的解决方法如下

pip3 install torch==1.3.0 torchvision==0.4.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

也就是安装1.3.0的torch和0.4.1的torchvision。

这就能用了,但是, torch.ops.torchvision.nms的输出还是RuntimeError: No such operator torchvision::nms。
不知道原因,最后附上pytorch官方github中同样问题的链接,(里面也没解释清),有同样问题的可以深入探讨一下具体的原因及更好的办法。

https://github.com/pytorch/vision/issues/1489
https://github.com/pytorch/vision/issues/1405

你可能感兴趣的:(pytorch)