程序员经验分享-hwhale

RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1

2023-10-26

报错：RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1
原因：在使用Pytorch加载模型时报错。加载的模型是用两个GPU训练的，而加载模型的电脑只有一个GPU，所以会出错。
解决：model = torch.load(model_path)
改为：model = torch.load(model_path, map_location=‘cuda:0’)
如果是4块到2块：就把map_location改为：map_location={‘cuda:1’: ‘cuda:0’}

在这里插入图片描述
自己根据报错的文件路径，一般先去找train文件，或者最简单的方法就是 关键词 torch.load
然后按全局搜索 ctrl+shift+f

本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)

python

深度学习

人工智能

RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1 的相关文章

随机推荐

  • log4j.properties 配置

    Logger级别 WARN 表明会出现潜在错误的情形 ERROR 指出虽然发生错误事件 但仍然不影响系统的继续运行 FATAL 指出每个严重的错误事件将会导致应用程序的退出 ALL 是最低等级的 用于打开所有日志记录 OFF 是最高等级的

  • 一文说清DC-DC BUCK电路（非常详细）

    目录 摘要 BUCK原理 DC DC芯片框图 自举电容 输出电感 输出电容和纹波 损耗 总结 摘要 DC DC BUCK 是硬件工程师工作中使用频率非常高的电路 可以这么说 只要板子不是迷你型的 十有八九都有DC DC 因此 对它的了解与学

  • 利用Python3做词频统计和词云图

    起源 因看到一篇满眼是字的文章 故希望能够快速的检索出关键字 所以尝试用Python3来实现 代码 import jieba import numpy import codecs import pandas import matplotli

  • Error-Java-IJ：Imported project refers to unknown jdks JavaSE-1.7

    ylbtech Error Java IJ Imported project refers to unknown jdks JavaSE 1 7 Import from EclipseImported project refers to u

  • uniapp 使用 axios

    1 先用npm install axios 这就不用说了吧 2 添加配置代码 App vue main js 引入封装后的axios import axios from utils http js 给Vue函数添加一个原型属性 axios

  • 【嵌入式实战】一文拿下 STM32 Lwip MQTT（超详细）

    文章目录 原创声明 前言 一 MQTT 是什么 二 Cube 配置 2 1 STM32 ETH 设置 2 2 修改 PHY 地址 2 3 LWIP 设置 在这里插入图片描述 三 生成工程的简单测试 3 1 手动修改 MAC 地址 3 2 P

  • 网络爬虫之xpath提取文本时的乱码

    问题如下大红框标注部分 解决方案 在请求获取后 加入转换即可解决 r encoding utf 8

  • 不懂技术，又不影响上班，网上的攒营汇适合做副业吗

    什么时候开始副业最好 铁饭碗 一工作就可以立即开始研究副业 社会人 毕业年后可以开始副业 最好是工资接近同龄人倍时 什么时候不适合开始副业 失业时 失业时主要任务是选对一个方向 先找一个稳定的工作发展 因为此时如果你开展副业 由于副业的不稳

  • STM32使用DMA接收串口数据

    目录 01 概述 02 DMA接收 03 中断 04 代码 01 概述 在之前的文章里 STM32串口详解 和 STM32 DMA详解 文章中 详细讲解了STM32的串口和DMA外设 本篇文章将不在细述串口和DMA的知识 在串口讲解的文章中

  • ajax的post请求数据为json格式相关设置

    平时用request封装axios习惯了突然用ajax传参时一直出问题 原来是配置设置错了 ajax type post contentType application json 必须项 dataType json 必须项 data JSO

  • 垃圾大学python小题

    Python程序设计 课程 实验报告四 字符串与正则表达式 文章如果太长请直接复制到sublime text 软件里进行关键字搜索 一 实验目的和要求 理解字符串编码格式 熟练掌握字符串的概念和各种用法 掌握Python标准库string的

  • 2020,会成为区块链在中国的元年吗？

    2020 会成为区块链在中国的元年吗 文章目录 2020 会成为区块链在中国的元年吗 区块链扫盲 区块链释义 区块链历史 区块链的特点 什么是中心化 什么是去中心化 程序员眼中的区块链 区块链的软分叉与硬分叉 以太坊与以太经典 金本位与银本

  • DDK下的Sleep函数KeDelayExecutionThread

    转载自 http www programlife net ddk kedelayexecutionthread like win32 sleep api html 许多读者一定使用过Sleep函数 这能使程序停下一段时间 许多需要连续 长期

  • TCP服务器/客户端实例(C/C++)

    本篇博客阅读有障碍请参阅 网络编程入门 1 1 Linux下的TCP服务器 include

  • Linux中find命令-path -prune用法详解(find命令如何只查找当前目录,而不查找当前目录下的子目录find . ! -name "." -type d -prune -o -type f -name "*.txt" -print)

    在Windows中可以在某些路径中查找文件 也可以设定不在某些路径中查找文件 下面用Linux中的find的命令结合其 path prune参数来看看在Linux中怎么实现此功能 假如在当前目录下查找文件 且当前目录下有很多文件及目录 多层

  • vue中的自定义指令

    一 什么是指令 指令系统是计算机硬件的语言系统 也叫机器语言 它是系统程序员看到的计算机的主要属性 因此指令系统表征了计算机的基本功能决定了机器所要求的能力 在vue中提供了一套为数据驱动视图更为方便的操作 这些操作被称为指令系统 我们看到

  • 蓝桥杯_算法训练_大小写转换

    问题描述 编写一个程序 输入一个字符串 长度不超过20 然后把这个字符串内的每一个字符进行大小写变换 即将大写字母变成小写 小写字母变成大写 然后把这个新的字符串输出 输入格式 输入一个字符串 而且这个字符串当中只包含英文字母 不包含其他类

  • Ubuntu——显示详细开机信息

    Ubuntu启动时默认是不显示详细信息的 现在想以字符界面方式启动系统 重启电脑 进入recovery模式 获得root权限后 做如下修改 1 vi etc default grub 注释掉下面一行 GRUB CMDLINE LINUX D

  • xray的使用&实现自动漏扫

    工具 下载地址 https github com chaitin xray releases 查看帮助文档 xray exe webscan h 基本使用 代理模式 xray的优点主要是被动扫描 除了http还需要代理https 生成ca证

  • RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1

    报错 RuntimeError Attempting to deserialize object on CUDA device 2 but torch cuda device count is 1 原因 在使用Pytorch加载模型时报错

热门标签