NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.

用户在观看B站视频时遇到YOLOv5运行错误,因torchvision版本与PyTorch不对应。通过检查发现torch使用了CUDA版本,而torchvision是CPU版本。解决方案是下载对应CUDA的torchvision包并正确安装,最终解决了运行时不报错的问题。
摘要由CSDN通过智能技术生成

看b站小土堆视频运行YOLOv5报错,百度查说是pytorch和torchvision版本不对应等问题,但我自己检查是对应的,所以不知道如何解决这个异常。

NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.
这是官网上版本对应的信息:

我的pytorch和torchvision版本:


后来看到是因为我的torchvision安装的是cpu版本,而torch是cuda版本的。找到对应版本的torchvision下载。download.pytorch.org/whl/torchvision/

 下载完放入anaconda的envs文件夹下,然后进行本地下载

重新查看版本:

再次运行YOLOv5,运行不报错:

参考文献:

Torchvision与Pytorch匹配的版本安装方法 - 知乎 (zhihu.com)

  • 8
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值