pytorch bug问题

2020-07-15  本文已影响0人  chunleiml

1、RuntimeError: No such operator torchvision::nms
出问题版本 torch=1.4 torchvision=0.5
解决方法:
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

上一篇 下一篇

猜你喜欢

热点阅读