【记个笔记】torchvision版本需要升高一点

最新推荐文章于 2024-06-17 23:04:19 发布
最新推荐文章于 2024-06-17 23:04:19 发布
阅读量401 收藏 7
点赞数 4
文章标签： 笔记
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_53383206/article/details/136891650
版权
本文讲述了在使用PyTorch时遇到的RuntimeError，关于torchvision::nms函数在CUDA后端无法运行的问题，以及提供了解决方案，特别是针对Facebook员工在移动设备上使用PyTorch的情况.
摘要由CSDN通过智能技术生成

RuntimeError: Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend. This could be because the operator doesn’t exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. ‘torchvision::nms’ is only available for these backends: [CPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, Tracer, Autocast, Batched, VmapMode].

CPU: registered at C:\Users\circleci\project\torchvision\csrc\ops\cpu\nms_kernel.cpp:111 [kernel]
BackendSelect: fallthrough registered at …\aten\src\ATen\core\BackendSelectFallbackKernel.cpp:3 [backend fallback]
Named: registered at …\aten\src\ATen\core\NamedRegistrations.cpp:7 [backend fallback]
AutogradOther: fallthrough registered at …\aten\src\ATen\core\VariableFallbackKernel.cpp:35 [backend fallback]
AutogradCPU: fallthrough registered at …\aten\src\ATen\core\VariableFallbackKernel.cpp:39 [backend fallback]
AutogradCUDA: fallthrough registered at …\aten\src\ATen\core\VariableFallbackKernel.cpp:43 [backend fallback]
AutogradXLA: fallthrough registered at …\aten\src\ATen\core\VariableFallbackKernel.cpp:47 [backend fallback]
Tracer: fallthrough registered at …\torch\csrc\jit\frontend\tracer.cpp:999 [backend fallback]
Autocast: fallthrough registered at …\aten\src\ATen\autocast_mode.cpp:250 [backend fallback]
Batched: registered at …\aten\src\ATen\BatchingRegistrations.cpp:1016 [backend fallback]
VmapMode: fallthrough registered at …\aten\src\ATen\VmapModeRegistrations.cpp:33 [backend fallback]

一只程序猿林
关注
  • 4
    点赞
  • 踩
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
torchvision nms
jacke121的专栏
06-21 6053
if __name__ == '__main__': import torch a=torch.Tensor([[1,1,2,2],[1,1,3.100001,3],[1,1,3.1,3]]) b=torch.Tensor([[0.9],[0.98],[0.980005]]) from torchvision.ops import nms ccc=nms(a,b,0.4) print(ccc) print(a[ccc])
ediary记笔记.7z
09-06
标题中的"ediary记笔记.7z"表明这是一个关于记笔记的应用或软件的...综上所述，"ediary记笔记.7z"不仅是一个压缩文件，它代表了记笔记软件的使用、数据备份、文件压缩与解压技术、软件版本管理等多个IT领域的知识点。
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.”
yolov5训练报错：NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ back
自成背后的博客
06-29 1256
NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' back
一、 Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend解决方法.
最新发布
monkey_lc的博客
06-17 1004
方法一：上述链接中内容比较详细，我这里简要提一下，在第一步框选的路径下，创建新的文件夹，内容放在文章末尾，然后命名xx（随意），然后在上图return返回值部分，改为 return xx.nms就行了（但是我自己的这样改，还是报错，没有找到声明，我不知道如何解决qaq，浪费了n小时时间，所以有了第二个方案）lt = torch.max(boxes1[:, None, :2], boxes2[:, :2]) # [N,M,2] # N中一个和M个比较；# 得分最大框对应的索引, 以及对应的坐标。
深度剖析问题：Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
Flying Bulldog 的博客
05-21 9886
（2）根据后台提示的信息，分析源码，进行问题的解决，下面是解决问题的过程：分析代码行：return torch.ops.torchvision.nms(boxes, scores, iou_threshold)，ctrl + 鼠标左键点击 nms，进入到下一个文件，截图如下图所示：在函数nms内部做一个标志（print(22222222222222)），观察能否调用函数nms，经测试发现没有调用函数nms。
10小石潭记笔记.pdf
03-11
初二小石潭记笔记
ubuntu平台为知笔记软件包，免安装版本
06-27
ubuntu平台为知笔记软件包，免安装版本。 基于ubuntu平台，下载后即可直接使用。
9桃花源记笔记.pdf
03-11
初二语文 桃花源记笔记
笔记记录分享-笔记记录分享网站-笔记记录分享网站源码-笔记记录分享网站java代码-基于Web的笔记记录分享网站设计与实现-代码
02-01
笔记记录分享-笔记记录分享网站-笔记记录分享网站源码-笔记记录分享网站java代码-笔记记录分享网站设计与实现-基于springboot的笔记记录分享网站-基于Web的笔记记录分享网站设计与实现-笔记记录分享项目-笔记记录...
NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.
weixin_43501408的博客
12-24 8499
NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.
CPU部署jit模型时报错：Could not run ‘atenempty_strided’ with arguments from the ‘CUDA’ backend
asd123pwj的博客
08-11 3428
这个问题网上很多说的是移动端部署的，但我这个是cpu部署，我感觉这问题应该很多人碰到才对，结果没搜到。没想到昨天才因为鸽了太久发了水贴，第二天就有帖子发了。，无图/无目录/格式错误/更多相关请至首发页查看。...
“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.”
qq_42548064的博客
05-30 4049
当前环境： CUDA-11.3 torch-1.11.0 torchvision-0.10.0+cpu 报错原因： 在官网下的torchvision是cpu版本：torchvision-0.10.0+cpu，不能调用CUDA 尝试解决： 在官网用conda重新安装 但是安装torchvision仍然是cpu版本。折腾几番，发现是因为torch-1.11.0还没有对应torchvision的gpu版本。 解决方案： 换成CUDA 11.3 对应的更低版本的torch与torchvision pip3 ins
安装高版本Pytorch以及torchvision时提示找不到指定版本的包报错解决
ShakalakaPHD的博客
09-17 6829
安装高版本Pytorch以及torchvision问题描述二级目录三级目录 问题描述 在使用Pytorch自带的faster RCNN时出现以下报错： RuntimeError: No such operator torchvision::nms 经过查找问题，发现是Pytorch版本与torchvision版本不一致导致的 但是在安装指定版本的Pytorch与torchvision时会出现报错： Could not find a version that satisfies the requiremen
笔记记录分享网站设计与实现-Java-SpringBoot
网站实现了多个关键模块，如收货地址管理、笔记广场、笔记收藏、留言管理、购物车、字典表、公告信息、关注管理、日常任务、领取任务、商品管理、商品收藏、商品评价、商品兑换、用户管理和管理员管理等。...

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值