小白初学YOLOv8 NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ back

已于 2023-10-26 14:38:39 修改
阅读量729 收藏
点赞数
文章标签： YOLO 深度学习 pytorch
于 2023-10-25 22:07:57 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/lpls_demon/article/details/134044724
版权

在本地跑yolov8模型遇见的一个报错，查了网上很多说法是torch版本和cuda版本的问题，目前我的GPU cuda版本是12.1，之前用的torch 12.1和torchvision 0.16+cpu（当时没意识到torchvision版本不支持cuda）会报错。

后续我降低了torch和torchvision的版本，分别是11.1和0.16+cu112，但还是会报错。在此记录一下，后续找到解决方法会更新。

 

后续终于解决了，降低版本后需要重启电脑，虽然不知道为什么重启电脑才能生效，但总归是能跑起来了

读书人271
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
YOLOv4-object：我们修改NMS procecss
02-10
YOLOv4-object：对象发现的有效模型和方法 此仓库基于 。抽象对象发现是指识别图像中的所有未知对象，这对于机器人系统探索未知环境非常重要。 近年来，基于深度学习方法的物体检测模型在物体分类和定位方面取得了令...
NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
weixin_43501408的博客
12-24 8137
NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.
[YOLOv8] 问题分析和解决 - NotImplementedError: Could not run ‘torchvision::nms‘ ...from the ‘CUDA‘ backend
老狼工作室的博客
12-14 2637
本文分析并解决了使用YOLOv8+Cuda训练数据集时抛出的错误 “NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. ”
YOLOv5运行问题：Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
weixin_43497556的博客
05-13 1004
解决YOLOv5的Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.问题
YOLOv5 报错：“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ back
Dontla的博客
08-28 1万+
是torchvisiao-cuda没装好，重装下 参考文章：解决“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.”
Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend的解决；用GPU跑Detectron2的demo
weixin_41793473的博客
07-08 1万+
Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend的解决；用GPU跑Detectron2的demo 在上一篇文章中，有一个问题没有解决。 Detectron2-入门1-极链AI云平台部署-多图好玩测试 Traceback (most recent call last): File "demo.py", line 112, in <module> predictions, visualized_
YoloV8改进策略：将CIoU替换成Wise-IoU，幸福涨点，值得拥有，还支持EIoU、GIoU、DIoU、SIoU无缝替换
08-04
YoloV8改进策略：将CIoU替换成Wise-IoU，幸福涨点，值得拥有，还支持EIoU、GIoU、DIoU、SIoU无缝替换
yolov3：PyTorch中的YOLOv3> ONNX> CoreML> TFLite
02-03
** GPU速度使用批处理大小为32的V100 GPU测量超过5000张COCO val2017图像的平均每张图像的端到端时间，包括图像预处理，PyTorch FP16推理，后处理和NMS。 来自EfficientDet数据（批量大小为8）。 预训练的检查站 ...
yolov5：PyTorch中的YOLOv5> ONNX> CoreML> TFLite
02-03
** GPU速度使用批处理大小为32的V100 GPU测量超过5000张COCO val2017图像的平均每张图像的端到端时间，包括图像预处理，PyTorch FP16推理，后处理和NMS。 来自EfficientDet数据（批量大小为8）。 2021年1月5日： ...
YOLOv8 问题记录
一点回忆
11-09 259
升级到最新的tensorflow，保证每个模块都是最新的。安装合适版本的pytorch。
yolov5训练报错：NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ back
自成背后的博客
06-29 1068
NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' back
NotImplementedError
热门推荐
kking_edc的博客
10-31 2万+
在python定义类的时候可以预先留出一个接口不实现，而在后续继承的子类中实现，当我们想要提醒自己这个类的子类一定要实现这个接口时，可以调用NotImplementError，具体使用方法如下所示： 注意当取消注释的时候调用成功 ...
【Debug记录】Pytorch搭建神经网络时报错‘NotImplementedError’
Roaddd的博客
12-12 1127
raise NotImplementedError
在运行yolo时的一些报错及解决方法
weixin_43834466的博客
12-12 4640
yolo,pytorch等一些报错个人汇集
tensorflow模型训练出现ModuleNotFoundError和NotImplementedError
zwl613613613的博客
06-02 477
tensorflow模型训练出错
YoloV8改进策略：BackBone改进DCNv4
qq_53545309的博客
05-08 120
涨点效果：在我自己的数据集上，mAP50 由0.986涨到了0.993，mAP50-95由0.737涨到0.77，涨点明显！DCNv4是可变形卷积的第四版，速度和v3相比有了大幅度的提升，但是环境搭建有一定的难度，对新手不太友好。如果在使用过程遇到编译的问题，请严格按照我写的环境配置。
西红柿叶病害检测（yolov8模型，从图像、视频和摄像头三种路径识别检测，包含登陆页面、注册页面和检测页面）
qq_40840797的博客
05-08 269
西红柿叶病害检测（yolov8模型，从图像、视频和摄像头三种路径识别检测，包含登陆页面、注册页面和检测页面）
mamba复现—mamba+yolov8魔改（win）
最新发布
qq_43767886的博客
05-11 970
将Mamba结合到Yolov8架构中的方法，旨在展示其在目标检测任务中的潜力。通过结合Mamba的优势，Yolov8_MambaCBAM旨在改善长距离信息捕获和全局建模能力，以提高目标检测任务的性能...
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、付费专栏及课程。

余额充值