极链ai云使用RTX3090报错：RuntimeError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA

最新推荐文章于 2024-03-17 10:58:08 发布
最新推荐文章于 2024-03-17 10:58:08 发布
阅读量1.2k 收藏 2
点赞数 1
分类专栏： 深度学习小问题 文章标签： 人工智能 深度学习 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/A__MP/article/details/125540864
版权

使用极链ai云配置好的Pytorch框架跑Yolov5的时候，会出现RuntimeError: Could not run ‘torchvision::nms’ with arguments from the 'CUDA的错误。
在这里插入图片描述
经过查询，是因为torch、torchvision和CUDA版本的问题，后来使用该文章中添加链接描述的配置解决了问题。（自己选用的是CUDA11.0，刚好是一致的！！）

# CUDA 11.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
小小的学徒
关注
  • 1
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
eclipse常用插件安装
04-11
文档有图有描述可以很清楚的看到，也很容易看懂，更好上手。
深度学习error记录（自用|持续更新）
Alannah的个人学习记录
05-20 1395
目录1.RuntimeError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. 1.RuntimeError: Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend. Traceback (most recent call last): File "/opt/yyl/yolov5/train.py", line
NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend. 错误纠正
最新发布
qq_61776376的博客
03-17 1967
但是NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend这个错误是由cup版本的torchvision引起的，我看了好多博主说是torch与torchvision版本不对的报错，接下来我说下我是怎么处理的。cu121表示cuda的版本号，我记得我的是12.2没有可以直接用121也行，进入这个页面，ctrl+f就可以自己搜到自己的版本号了。查看自己的版本号，右上角有个。
深度剖析问题：Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
Flying Bulldog 的博客
05-21 9390
（2）根据后台提示的信息，分析源码，进行问题的解决，下面是解决问题的过程：分析代码行：return torch.ops.torchvision.nms(boxes, scores, iou_threshold)，ctrl + 鼠标左键点击 nms，进入到下一个文件，截图如下图所示：在函数nms内部做一个标志（print(22222222222222)），观察能否调用函数nms，经测试发现没有调用函数nms。
成功解决：NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
m0_74890428的博客
05-31 9480
成功解决：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_
YOLOv5运行问题：Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
weixin_43497556的博客
05-13 1012
解决YOLOv5的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.
wyw0000的博客
01-17 1003
NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
failed call to cuInit: CUDA_ERROR_NO_DEVICE解决方法
01-06
在anaconda虚拟环境下运行程序时，报错failed call to cuInit: CUDA_ERROR_NO_DEVICE. 单独测试import tensorflow，可以成功调用GPU。 用ncvv -V，或nvidia-smi，都可以正常看到CUDA安装版本，及GPU使用情况。 问题...
rtx:Geforce RTX搜寻器:face_with_tears_of_joy::fire:
04-19
rtx Geforce RTX搜寻器 :face_with_tears_of_joy: :fire:
电子行业点评：英伟达推出Chat with RTX，AI PC生态再加速.pdf
03-04
半导体 电子行业 行业分析 数据分析 数据报告 行业报告
TARS:rtx 3090 go brr
03-18
塔斯 rtx 3090 go brr
RayTracingInVulkan:使用Vulkan和NVIDIA的RTX扩展实现Peter Shirley的《在一个周末进行光线追踪》一书
03-28
Vulkan中的光线追踪 我使用Vulkan和NVIDIA的RTX扩展（以前是VK_NV_ray_tracing，现在移植到Khronos跨平台VK_KHR_ray_tracing_pipeline扩展）实现了书中 。 这允许在适当的硬件上以交互速度渲染大多数场景。 实时光线...
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.”
NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend解决办法
编程研究坊的博客
03-13 4972
NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend解决办法
【python】yolov5的torch与torchvision环境问题
聿默的博客
05-15 7667
1.可用环境 torch==1.7.1+cu110 torchvision==0.8.2+cu110 2.错误环境遇到的问题 （1）情况一 torch==1.7.0+cu110 torchvision==0.8.0 训练yolov5时，训练就会开始报错： return torch.ops.torchvision.nms(boxes, scores, iou_threshold) RuntimeError: Could not run 'torchvision::nms' with argu
RuntimeError: CUDA error: misaligned address
09-05
引用中提到的错误"RuntimeError: CUDA error: misaligned address"是由于CUDA内核错误引起的。该错误可能会在其他API调用时异步报告，导致下面的堆栈跟踪可能是不正确的。为了调试，可以考虑在CUDA_LAUNCH_BLOCKING=1的情况下运行程序。 引用中提到的解决方法是通过设置CUDA_LAUNCH_BLOCKING=1来进行调试，以便在错误发生时阻塞程序并获得准确的堆栈跟踪。这样可以更容易地定位错误并解决问题。 引用中提到的错误"RuntimeError: CUDA out of memory"是由于GPU显存不足引起的。为了解决这个问题，可以尝试以下方法： 1. 将batch_size改小，以减小每次运算所需的显存占用量。 2. 当需要取torch变量的标量值时，使用item()属性而不是保留整个变量。 3. 在测试阶段，可以尝试添加代码以释放不必要的显存资源。 引用中提到的警告"UserWarning: NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation"表示当前的PyTorch版本不支持您的显卡。如果想要使用NVIDIA GeForce RTX 3090 GPU进行PyTorch开发，请参考提供的链接，按照指引进行安装和配置。 综上所述，"RuntimeError: CUDA error: misaligned address"是由CUDA内核错误引起的，可以使用CUDA_LAUNCH_BLOCKING=1进行调试。而"RuntimeError: CUDA out of memory"则是由于GPU显存不足引起的，可以通过减小batch_size、使用item()属性和释放不必要的显存资源来解决问题。同时，如果要使用NVIDIA GeForce RTX 3090 GPU进行PyTorch开发，请确保PyTorch版本与显卡兼容。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [CUDA Error： no kernel image is available for execution on device](https://blog.csdn.net/qq_34845880/article/details/126460123)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* [pytorch模型提示超出内存RuntimeError: CUDA out of memory.](https://download.csdn.net/download/weixin_38552305/14884908)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值