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

最新推荐文章于 2024-03-17 10:58:08 发布
最新推荐文章于 2024-03-17 10:58:08 发布
阅读量189 收藏 5
点赞数 8
文章标签： yolov5
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_45869597/article/details/136263216
版权
用户在观看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
    评论
Pytorch安装问题：pip3 install torchvision报错解决
03-28
在已有anaconda条件下再次安装新的anaconda遇到问题并解决。
第讲：C1单片机应用与C语言程序设计总纲ppt课件.ppt
11-15
接着展示了如何编写第一个程序，该程序通过包含`<uart.h>`头文件，利用`main()`函数和`printf()`函数向串口发送"Hello,this is a message from your Robot"的信息。`//`用于注释，不会被执行。 在硬件部分，课程...
成功解决：NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
m0_74890428的博客
05-31 1万+
成功解决：NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.
NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend解决办法
编程研究坊的博客
03-13 5273
NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend解决办法
YOLOX目标检测实战：训练自己的数据集
07-25
YOLOX是旷视科技新近推出的高性能实时目标检测网络，性能超越了YOLOv3/YOLOv4 /YOLOv5。YOLOX使用 PyTorch开发，采用了Anchor-free机制、解耦头、Multi Positives、先进的标签分配策略和强数据增广等前沿技术。 本课程将手把手地教大家使用labelImg标注和使用YOLOX训练自己的数据集，完成一个多目标检测实战项目，可检测图像和视频中的足球和梅西两个目标类别。  本课程分别在Windows和Ubuntu系统上做项目演示。包括：安装YOLOX、标注自己的数据集、准备自己的数据集（自动划分训练集和验证集）、修改配置文件、训练自己的数据集、测试训练出的网络模型和性能统计。  
NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend. 错误纠正
qq_61776376的博客
03-17 2917
但是NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend这个错误是由cup版本的torchvision引起的，我看了好多博主说是torch与torchvision版本不对的报错，接下来我说下我是怎么处理的。cu121表示cuda的版本号，我记得我的是12.2没有可以直接用121也行，进入这个页面，ctrl+f就可以自己搜到自己的版本号了。查看自己的版本号，右上角有个。
解决“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.”
热门推荐
weixin_51179975的博客
07-25 1万+
之前我采用的是离线方式分别下载了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/w
SnmpScan.rar_site:www.pudn.com_snmpscan_snmp协议工具_拓扑_拓扑图工具
07-14
SNMP工作在应用层，基于TCP/IP协议族，主要由管理信息库（MIB）、代理（Agent）和网络管理系统（NMS）三部分组成。通过查询MIB，NMS可以获取设备的各种运行参数和配置信息。 **网络拓扑**: 网络拓扑是指网络设备间...
nms:Node.js + Mocha.js + Server.js = NMS
06-04
Node.js + Mocha.js + Server.js = NMS 这是一个 shell 项目，旨在帮助您开始使用 Node.js 构建服务。 它还提供了一个结构和一组约定，使您能够快速上手并希望编写更清晰、更可测试的代码。背景开始使用 Node.js 很...
activemq-nms-stomp：Apache ActiveMQ NMS.Stomp客户端的镜像
02-05
Apache ActiveMQ NMS.Stomp客户端是一个用于连接到Stomp协议兼容消息代理的.NET实现，主要设计用于与Apache ActiveMQ交互。这个项目是Apache ActiveMQ的一部分，它提供了多种语言的客户端库，包括Java、Python、Ruby...
ASP外观专利图像检索平台(源代码+论文).rar
07-21
ASP外观专利图像检索平台(源代码+论文)
C++课程设计：电煤气管理系统【源码+文档】
07-21
C++课程设计：电煤气管理系统【源码+文档】 本程序是一个水电气管理信息系统，能够对高校的水电气费用进行管理, 包括了成员的基本信息，如学号、编号、姓名、成员水电气的用量等。程序的用途包括缴纳水电气费、查询一个同学水电气费用量、查看所有同学的缴费情况、增加学生信息、删除学生信息、退出系统等。在设计时也考虑到学生和教师在用水电气时的不同，学生可以免费使用一定额度的水电气，超过这个额度的以后必须付费，且付费部分水电气费的价格要高于教工的收费标准，该措施的实行是为了鼓励同学们节约资源，以免造成不必要的资源浪费。该软件主要是为了学校的管理人员提供便捷，以更快的完成水电气费用的收缴。该软件本着简洁明了，实用稳定为一体进行设计。 系统将实现以下功能： （1）实现对用户信息的录入； （2）实现水电煤气数据的录入； （3）实现计算并查询用户应缴费用，查询未缴纳费用的名单； （4）实现对人员的删除和添加；
ASP+ACCESS网上动态同学录系统（源代码+论文+系统说明+答辩PPT）.rar
07-21
ASP+ACCESS网上动态同学录系统（源代码+论文+系统说明+答辩PPT）
avaloniassification-mas笔记
最新发布
07-21
avalonia
ASP+ACCESS学生排课管理系统毕业设计(源代码+论文).rar
07-21
ASP+ACCESS学生排课管理系统毕业设计(源代码+论文)
ASP+ACCESS在线教育系统设计(源代码+论文).rar
07-21
ASP+ACCESS在线教育系统设计(源代码+论文)
NotImplementedError: Could not run torchvision::nms with arguments from the CUDA backend.
05-22
这个错误通常是因为您的CUDA版本与您安装的PyTorch版本不兼容导致的。您可以尝试使用匹配的CUDA版本重新安装PyTorch，或者尝试使用CPU运行代码，以避免该错误。您也可以尝试更新您的PyTorch和torchvision版本到最新的稳定版本来解决该问题。如果问题仍然存在，请检查您的CUDA是否正确安装并配置。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值