【onnx问题解决】关键词:found at least two devices、torch.onnx.export

关键词:Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0!

报错:

[34m[1mONNX:[0m export failure ❌ 3.8s: Expected all tensors to be on
the same device, but found at least two devices, cpu and cuda:0! (when
checking argument for argument other in method wrapper__equal)
[34m[1mTensorRT:[0m export failure ❌ 3.9s: Expected all tensors to
be on the same device, but found at least two devices, cpu and cuda:0!
(when checking argument for argument other in method wrapper__equal)
Traceback (most recent call last): File
“/home/nvidia/ZED2i/ros2_ws/src/export.py”, line 14, in
main() File “/home/nvidia/ZED2i/ros2_ws/src/export.py”, line 9, in main
model.export(format=‘engine’, opset=11) # creates ‘yolov8n.engine’ File
“/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/model.py”,
line 310, in export
return Exporter(overrides=args, _callbacks=self.callbacks)(model=self.model) File “/home/nvidia/.local/lib/python3.8/site-packages/torch/autograd/grad_mode.py”,
line 27, in decorate_context
return func(*args, **kwargs) File “/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/exporter.py”,
line 252, in call
f[1], _ = self.export_engine() File “/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/exporter.py”,
line 122, in outer_func
raise e File “/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/exporter.py”,
line 117, in outer_func
f, model = inner_func(*args, **kwargs) File “/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/exporter.py”,
line 586, in export_engine
f_onnx, _ = self.export_onnx() File “/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/exporter.py”,
line 122, in outer_func
raise e File “/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/exporter.py”,
line 117, in outer_func
f, model = inner_func(*args, **kwargs) File “/home/nvidia/.local/lib/python3.8/site-packages/ultralytics/engine/exporter.py”,
line 333, in export_onnx
torch.onnx.export( File “/home/nvidia/.local/lib/python3.8/site-packages/torch/onnx/init.py”,
line 319, in export
return utils.export(model, args, f, export_params, verbose, training, File
“/home/nvidia/.local/lib/python3.8/site-packages/torch/onnx/utils.py”,
line 113, in export
_export(model, args, f, export_params, verbose, training, input_names, output_names, File
“/home/nvidia/.local/lib/python3.8/site-packages/torch/onnx/utils.py”,
line 737, in _export
params_dict = torch._C._jit_pass_onnx_deduplicate_initializers(graph, params_dict,
RuntimeError: Expected all tensors to be on the same device, but found
at least two devices, cpu and cuda:0! (when checking argument for
argument other in method wrapper__equal)

在这里插入图片描述

do_constant_folding=False,  # WARNING: DNN inference with torch>=1.12 may require do_constant_folding=False

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/diannao/17234.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Amazon云计算AWS之[7]内容推送服务CloudFront

文章目录 CDNCDN简介CDN网络技术 CloudFrontCloudFront基本概念 CDN CDN简介 用户在发出服务请求后,需要经过DNS服务器进行域名解析后得到所访问网站的真实IP,然后利用该IP访问网站。在这种模式中,世界各地的访问者都必须直接和网站服务器连…

openflow协议抓包分析

1、准备实验拓扑: 在Mininet环境中创建一个简单的SDN拓扑,包括控制器、交换机、主机等。 确保拓扑能够正常运行,SDN交换机与控制器建立连接。 采用主机Ubuntu22.04主机,IP地址是192.168.87.130,安装opendaylight控制…

c语言如何格式化读取文件内容

c语言中&#xff0c;格式化读取文件内容需要用到fscanf语句 它的语法格式是fscanf(文件指针&#xff0c;格式字符串&#xff0c;输入表列&#xff09; 示例代码如下 假如有一个eee.txt的文件&#xff0c;它的内容是123和3.1415 #include<stdio.h>int main() {FILE *f…

Git标签管理

文章目录 1. 什么是标签2. 创建标签3. 标签删除4. 本地标签推送至远程5. 标签远程删除 1. 什么是标签 标签tag &#xff0c;可以简单的理解为是对某次commit的⼀个标识&#xff0c;相当于起了⼀个别名。 例如&#xff0c;在项目发布某个版本的时候&#xff0c;针对最后一次co…

PG实践|PostgreSQL的安装和配置

&#x1f4eb; 作者简介&#xff1a;「六月暴雪飞梨花」&#xff0c;专注于研究Java&#xff0c;就职于科技型公司后端工程师 &#x1f3c6; 近期荣誉&#xff1a;华为云云享专家、阿里云专家博主、腾讯云优秀创作者、ACDU成员 &#x1f525; 三连支持&#xff1a;欢迎 ❤️关注…

代码随想录算法训练营第20天 |● 654.最大二叉树 ● 617.合并二叉树 ● 700.二叉搜索树中的搜索 ● 98.验证二叉搜索树

文章目录 前言654.最大二叉树思路方法一 递归法方法一2 老师的优化递归法 617.合并二叉树思路方法一 递归法方法二 迭代法 700.二叉搜索树中的搜索思路方法一 递归法方法二 迭代法 98.验证二叉搜索树思路方法一 使用数组方法二 不使用数组代码注意点&#xff1a; 方法二 使用双…

攻击者常用的五个数据中转网站

近来&#xff0c;各种数据中转网站被攻击者广泛用于传播代码片段、配置文件和各种文本数据&#xff0c;尽管这为研究人员提供了观察的窗口&#xff0c;但敏感信息被上传到互联网上时&#xff0c;也会对受害者构成巨大威胁。 这些网站通常并不需要注册或者身份验证&#xff0c;…

SD-WAN和IPSEC组网的区别

SD-WAN和IPSEC网络是现代企业网络中两种常见的解决方案。虽然它们在网络连接和安全方面发挥着重要作用&#xff0c;但在设计和功能上存在一些显著差异。 I. SD-WAN的概述和优势 SD-WAN代表软件定义广域网&#xff0c;通过集中管理和控制网络连接&#xff0c;提供更灵活、可靠、…

深入了解Linux中的环境变量

在Linux系统中&#xff0c;环境变量&#xff08;Environment Variables&#xff09;是用于配置操作系统和应用程序运行环境的一种机制。它们储存在键值对中&#xff0c;可以控制程序的行为、路径查找和系统配置。本文将深入探讨环境变量的基本概念、常见类型、设置和管理方法&a…

Latex公式编辑:在矩阵内画横线与竖线

在LaTeX中&#xff0c;要在矩阵内绘制横线和竖线&#xff0c;我们通常使用array或matrix环境&#xff0c;并结合\hline&#xff08;用于横线&#xff09;和|&#xff08;用于竖线&#xff09;来实现。但需要注意的是&#xff0c;\hline通常用于表格环境中。 LaTeX中绘制分块矩阵…

idea配置ssh、sftp连接服务器,docker插件使用,极其方便,无需再开第三方软件去操作服务器了,集成用于Idea一体

目录 配置SSH连接服务器 配置SFTP文件传输 Docker插件使用 配置SSH连接服务器 Setting>Tools>SSHConfigurations,点击加号新建一个连接&#xff0c;认证形式选key pair 打开终端&#xff0c;连接服务器&#xff0c;就可以终端操作服务器了 配置SFTP文件传输 文件浏览窗…

ThreadLocal简介

Thread类中&#xff0c;有个ThreadLocal.ThreadLocalMap 的成员变量。 ThreadLocalMap内部维护了Entry数组&#xff0c;每个Entry代表一个完整的对象&#xff0c;key是ThreadLocal本身&#xff0c;value是ThreadLocal的泛型对象值 public void set(T value) {Thread t Thread…

装机必备——WinRAR安装教程

装机必备——WinRAR安装教程 软件下载 软件名称&#xff1a;WinRAR 软件语言&#xff1a;简体中文 软件大小&#xff1a;3.38M 系统要求&#xff1a;Windows7或更高&#xff0c; 32/64位操作系统 硬件要求&#xff1a;CPU2GHz &#xff0c;RAM4G或更高 下载通道①迅雷云盘丨下…

Linux线程:线程控制

目录 一、线程的退出与等待 1.1pthread_join线程等待 1.2线程异常 1.3线程如何退出和结束 ​编辑 二、线程切换 三、线程的优缺点 3.1优点 3.2缺点 3.3线程vs进程 四、多线程的使用及实操 4.1堆空间共享 一、线程的退出与等待 在Linux中线程具有如下的特点&#xf…

postman教程-4-发送post请求

领取资料&#xff0c;咨询答疑&#xff0c;请➕wei: June__Go 上一小节我们学习了postman发送get请求的方法&#xff0c;本小节我们讲解一下postman发送post请求的方法。 POST请求通常用于向服务器提交数据以创建新资源或执行某些操作。与GET请求不同&#xff0c;POST请求可…

景源畅信电商:做抖店的成本高吗?

在当今数字化时代&#xff0c;抖音不仅仅是一个分享短视频的平台&#xff0c;更是一个充满商机的市场。随着抖音用户量的激增&#xff0c;越来越多的商家和个人开始关注在抖音上开设店铺&#xff0c;即所谓的“抖店”。那么&#xff0c;做抖店的成本高吗?这个问题困扰着许多初…

【鱼眼镜头11】Kannala-Brandt模型和Scaramuzza多项式模型区别,哪个更好?

Kannala-Brandt模型和Scaramuzza多项式模型在描述鱼眼相机畸变时都有其特定的数学表示和应用&#xff0c;但它们之间存在一些区别。以下是对两者区别的分点表示和归纳&#xff1a; 数学表示&#xff1a; Kannala-Brandt模型&#xff1a;它假设图像光心到投影点的距离和角度的多…

Autodl如何进行实例使用(同区)

一、首先找到之前保存的实例 二、点击更多然后选择克隆实例 三、选择是否要保存之前的数据盘 四、选择空余的GPU进行创建即可

【Redis】持久化操作详解

Redis 持久化操作详解 Redis 实现持久化的时候&#xff0c;具体是按照什么样的策略来实现的呢&#xff1f; Redis支持两种方式的持久化&#xff0c;一种是RDB方式、另一种是AOF&#xff08;append-only-file&#xff09;方式&#xff0c;两种持久化方式可以单独使用其中一种&…

TypeScript详解

文章目录 一.TS 基础概念1.什么是TS2.TS基础类型与写法* boolean string number array null undefined* tuple - 元组* enum - 枚举* any unknown void* object &#xff5c; Object &#xff5c; {} - 对象 二. 接口 - interface三.交叉类型四.断言 - 类型声明&#xff0c;转换…