解决“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.”

之前我采用的是离线方式分别下载了pytorch以及torchvision，由于版本不匹配，出现了这个报错。卸载之后到官网（[https://pytorch.org]）下载，成功解决。
下面是采用pip的方式下载pytorch的命令。由于文件比较大，我采用了豆瓣源下载的方式，这样速度会比较快。

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

训练脚本运行成功！