RuntimeError: No such operator torchvision::nms问题解决
作者今天在跑之前写过的一个算法时遇到了这个问题。torch和torchvision版本和之前用的时侯没有变化,torch==1.4.0+cu100 torchvision==0.5.0+cu100。并且import torchtorch.ops.torchvision.nms输出了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
更多推荐
所有评论(0)