成功解决: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包

成功解决:NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend._第1张图片

下载对应的包放在对应的项目的目录下,进入该项目的环境中进行pip 安装

pip install torchvision-0.11.1+cu113-cp36-cp36m-win_amd64.whl

 

 

你可能感兴趣的:(报错,数学建模,python,pytorch)