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

原因:
1、torch和torchvision版本不同,比如torch是支持cuda的,但是torchvision不支持cuda
2、torch和cuda版本不对
解决方式:
1、卸载torch或者torchvision,重新安装 相同版本的torch和torchvision
2、卸载torchvision或者cuda,并分别重新安装,使版本匹配
我查看了自己pytorch环境下了的torch和torchvision的两个版本,发现这个两个依赖的版本不一样,对于torch来说,它是支持CUDA版本的,在下面的图片中也可以看出;而torchvision并不支持CUDA所以需要重新安装一下,使他们的版本对应。
![]()
重新安装torchvision, 点击安装pytorh或torchvision包

下载对应的包放在对应的项目的目录下,进入该项目的环境中进行pip 安装
pip install torchvision-0.11.1+cu113-cp36-cp36m-win_amd64.whl