解决RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cp

Python79

今天在把.pt文件转ONNX文件时,遇到此错误。

报错

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 mat2 in method wrapper_mm)

原因

代码中的Tensor,一会在CPU中运行,一会在GPU中运行,所以最好是都放在同一个device中执行。

pytorch有两种模型保存方式

一、保存整个神经网络的的结构信息和模型参数信息,save的对象是网络net

二、只保存神经网络的训练模型参数,save的对象是net.state_dict()

对应两种保存模型的方式,pytorch也有两种加载模型的方式。对应第一种保存方式,加载模型时通过torch.load('.pth')直接初始化新的神经网络对象;对应第二种保存方式,需要首先导入对应的网络,再通过net.load_state_dict(torch.load('.pth'))完成模型参数的加载。

输入验证码查看隐藏内容

扫描二维码关注本站微信公众号 Johngo学长
或者在微信里搜索 Johngo学长
回复 svip 获取验证码
wechat Johngo学长

相关文章
Python

变量和简单数据类型

2.1变量的命名和使用: 变量名只能包含字母、数字和下划线。不能包含空格。 不要将python关键字和函数名作为变量名 应使用小写的python变量名 x,y,z=1,2,3 2.2字符串的处理(用引...
Python

python os模块详细用法

os 模块提供了非常丰富的方法用来处理文件和目录 # 获取当前路径 path = os.getcwd() # 获取当前绝对路径 os.path.abspath(path) # 创建一级目录 os.mk...
Python

Python 输入/输出

程序要实现人机交互功能,需要能够向显示设备输出有关信息及提示,同时也要能够接收从键盘输入的数据。Python提供了用于实现输入/输出功能的函数input( )和输出函数 print(),下面分别对这两...
Python

python—关闭 close,quit

python---关闭 close,quit 原创 夕陌2022-07-19 11:26:02©著作权 文章标签 导包 firefox 文章分类 Python 编程语言 ©著作权归作者所有:来自51C...
Python

python自动化办公

一,xlwings的安装与使用 1,xlwings是什么 在日常生活中我们或多或少的都会跟Excel打交道,比如做销售统计,人力的考勤,学生的考试成绩等等,甚至在某些领域会涉及到批量操作Excel表格...
Python

python之文件处理

前戏 应用程序运行过程中产生的数据最先都是存放于内存中的,若想永久保存下来,必须要保存于硬盘中。应用程序若想操作硬件必须通过操作系统,而文件就是操作系统提供给应用程序来操作硬盘的虚拟概念,用户或应用程...
Python

Python 转义字符、格式化字符

Python 转义字符、格式化字符 精选原创 lanxiaofang2022-08-22 23:25:28博主文章分类:Python ©著作权 文章标签 python 转义 格式化 字符 文章分类 P...
Python

7 天学个Go,Go 结构体 + Go range 来学学

写在学习前 在前一篇博客我们学习了 Go 数组,其要求所有元素为同一数据类型,如果希望存储不同类型的数据,就要用到结构体相关知识。 结构体的定义:存储相同或不同类型的数据集合。 有 C 相关经验,结构...
Python

Python 协程并发下载斗鱼美女图片

仅用学习参考,转载请注明出处 爬虫技术第一步 作为爬虫技术第一步,就是可以使用爬取下载图片作为示例。 那么爬取图片,我们肯定去爬取美女图片作为第一步操作示例呀。 那么下面一个最重要的工作。 什么是最重...
Python

初识Python系列(一)

对于Python selenium操作的总结(一) 1.对于驱动的安装 驱动包:webdriver(在cmd执行help(webdriver)可查看所支持的浏览器类型,在此只提其中的Chrome和Ed...