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