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

news2025/5/22 20:58:06

关键词: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.coloradmin.cn/o/1703946.html

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

相关文章

【R语言】获取任意颜色的HTML 颜色代码、十六进制颜色代码、 RGB代码

网站来源: https://htmlcolorcodes.com/ 界面如下所示: 通过鼠标任意选择不同的颜色,就能获取该色的十六进制代码、RGB代码等。 除此之外,还提供了一些常用颜色的便捷选项,如下: 任意选择一种颜色,即可出…

Java开发快速入门

Java执行流程分析 .java文件 **(源文件)** ---javac编译--->.class文件 **(字节码文件)** ---java运行--->结果 运行的本质: .java文件称之为源文件 .class文件称之为字节码文件 什么是编译 javac Hello.java 1. 有了java源文件, 通过编译器将其编译成JVM可以识别的…

【吊打面试官系列】Java高并发篇 - AQS 支持几种同步方式 ?

大家好,我是锋哥。今天分享关于 【AQS 支持几种同步方式 ?】面试题,希望对大家有帮助; AQS 支持几种同步方式 ? 1、独占式 2、共享式 这样方便使用者实现不同类型的同步组件,独占式如 ReentrantLock&…

C语言动态顺序表结构的创建、初始化结构、尾插、尾删、头插、头删、指定位置插入、指定位置删除、找指定数值下标等的介绍

文章目录 前言一、 结构创建二、 初始化结构三、 打印动态顺序表四、 销毁动态顺序表五、 尾插六、尾删七、 头插八、 头删九、指定位置插入十、指定位置删除十一、找指定数值下标总结 前言 C语言动态顺序表结构的创建、初始化结构、尾插、尾删、头插、头删、指定位置插入、指…

Java 实验12 线程同步与通信

(一)实验目的 1、掌握JAVA中多线程的实现方法; 2、重点掌握多线程的同步与通信机制; 3、熟悉JAVA中有关多线程同步与通信的方法 ; 4、能使用多线程机制解决实际应用中的线程同步与通信问题。 (二&…

【社会信用体系1003】 企业违规新解:社会信用环境改善的实证分析!

今天给大家分享的是来自于国内顶级期刊金融研究2023年发表论文——《社会信用环境改善降低了企业违规吗?——来自“中国社会信用体系建设”的证据》所用到的重要数据集,该文章从企业层面探讨了社会信用系统建设对企业违规行为的影响,更精准地…

Srping 历史

一、History of Spring and the Spring Framework Spring came into being in 2003 as a response to the complexity of the early J2EE specifications. While some consider Java EE and its modern-day successor Jakarta EE to be in competition with Spring, they are …

小白不知道怎么投稿?记住这个好方法

作为一名单位信息宣传员,我最初踏上这条道路时,满心憧憬着通过文字传递我们单位的精彩瞬间,让社会听见我们的声音。然而,理想与现实之间的距离,却在一次次邮箱投稿的石沉大海中渐渐清晰。那时的我,像所有“小白”一样,以为只要用心撰写稿件,通过电子邮件发给各大媒体,就能收获满…

亚马逊云主管马特·加尔曼面临压力,致力于在人工智能领域赶超竞争对手

每周跟踪AI热点新闻动向和震撼发展 想要探索生成式人工智能的前沿进展吗?订阅我们的简报,深入解析最新的技术突破、实际应用案例和未来的趋势。与全球数同行一同,从行业内部的深度分析和实用指南中受益。不要错过这个机会,成为AI领…

Learning Deep Intensity Field for Extremely Sparse-View CBCT Reconstruction

摘要 稀疏视图锥束CT(CBCT)重建是降低辐射剂量和临床应用的一个重要方向,以往基于体素的重建方法将CT表示为离散体素,由于使用3D解码器,导致存储要求高和空间分辨率有限。我们将CT体积表示为连续强度场,并…

全面解析Java.lang.ClassCastException异常

全面解析Java.lang.ClassCastException异常 全面解析Java.lang.ClassCastException异常:解决方案与最佳实践 🚀📚摘要引言1. 什么是Java.lang.ClassCastException?代码示例 2. 报错原因2.1 类型不兼容2.2 泛型类型擦除2.3 接口和实…

【C++】STL快速入门基础

文章目录 STL(Standard Template Library)1、一般介绍2、STL的六大组件2.1、STL容器2.2、STL迭代器2.3、相关容器的函数vectorpairstringqueuepriority_queuestackdequeset, map, multiset, multimapunordered_set, unordered_map, unordered_multiset, …

Autodesk Flame 2025 for Mac:视觉特效制作的终极利器

在数字时代,视觉特效已经成为电影、电视制作中不可或缺的一部分。Autodesk Flame 2025 for Mac,这款专为视觉特效师打造的终极工具,将为您的创作提供无尽的可能。 Autodesk Flame 2025 for Mac拥有强大的三维合成环境,能够支持您…

stm32通过esp8266连接阿里云平台代码讲解

连接服务器 首先,按照一定的规则,获取连接阿里服务器所需要的ClientID(客户端D)、Username(用户名)、Passward(密码),ServerIP(域名),ServerPort&#xff08…

Qt | QTabBar 类(选项卡栏)

01、上节回顾 Qt | QStackedLayout 类(分组布局或栈布局)、QStackedWidget02、简介 1、QTabBar类直接继承自 QWidget。该类提供了一个选项卡栏,该类仅提供了一个选项卡, 并没有为每个选项卡提供相应的页面,因此要使选项卡栏实际可用,需要自行为每个选项卡设置需要显示的页…

冯 • 诺依曼体系结构和操作系统

目录 冯诺依曼体系结构基于冯诺依曼体系数据的高效流转数据流转示例操作系统(Operator System)操作系统(Operator System)层次结构硬件部分系统软件部分用户部分 管理——先描述,再组织 就一个程序而言,需要在计算机中运行的才能实现它的价值&#xff0c…

jmeter发送webserver请求和上传请求

有时候在项目中会遇到webserver接口和上传接口的请求,大致参考如下 一、发送webserver请求 先获取登录接口的token,再使用cookie管理器进行关联获取商品(webserver接口),注意参数一般是写在消息体数据中,消息体有点像HTML格式 执…

调整GIF图大小的方法是什么?分享4个

调整GIF图大小的方法是什么?在数字化时代,GIF以其独特的动图魅力,成为了网络交流中不可或缺的一部分。无论是社交媒体、博客文章还是工作汇报,一个恰到好处的GIF图往往能有效吸引观众的注意,传递信息,但过大…

【typescript - tsc 编译后路径问题/路径别名问题】

这几天在写typescript,遇到个路径依赖问题,编写的.ts文件直接运行OK,但是编译成.js后,运行提示 Error: Cannot find module xxx,📝记录分析和解决过程 。 问题描述 原始文件,有index.ts 其会引…

中小学校活动怎样投稿给媒体报道宣传?

身为一名学校老师,同时承担起单位活动向媒体投稿的宣传重任,我深知每一次校园活动背后的故事,都承载着师生们的辛勤汗水与教育的无限可能。起初,我满怀着对教育的热情,希望通过文字传递校园的温暖与光芒,却在投稿的道路上遇到了前所未有的挑战。 最初,我选择了最传统的路径——…