成功解决
RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1.

报错内容

程序在这一步报错
checkpoint = torch.load(‘model5_4.pt’)

以上问题描述是说未获取到当前环境下的 cuda，因为我的模型是在服务器上跑的，下载到本地后环境不同。

解决方法

若你当前在只有 CPU 环境下运行的话，需要加上map_location=torch.device(‘cpu’)。
若你当前在有 CUDA环境下运行的话，需要加上map_location=torch.device(‘cuda’)。
checkpoint = torch.load(‘model5_4.pt’)
即换成：

checkpoint = torch.load('model5_4.pt',map_location='cuda')

运行成功！不报错了！

RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1.相关推荐

  1. 【错误处理】RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count

    RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count 1 项目场景 2 ...

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

    问题描述: 运行测试程序的时候,报错"RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cu ...

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

    解决报错RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is ...

  4. RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is

    pytorch,pycharm下做项目遇到的一个小问题,报了如下错误: RuntimeError: Attempting to deserialize object on a CUDA device ...

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

    我是在服务器上训练的模型,然后下载到本地进行测试,报错如下: 问题应该是训练和测试的GPU不是同一个,需要将GPU改成本地的. 解决方法,在加载模型的语句上,增加 map_location='cuda ...

  6. PyTorch笔记： GPU上训练的模型加载到CPU/错误处理Attempting to deserialize object on a CUDA device but torch.cuda.is_a

    我之前在GPU上训练了一个模型,同时把模型的参数记录在resnet18_ultimate.pkl上 在本地的CPU上,我想把参数加载,于是一开始我是这么写代码的: import torch impor ...

  7. Attempting to deserialize object on a CUDA device but torch.cuda.is_available()的可能原因

    RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is Fal ...

  8. Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False

    Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False 此时改为: torch ...

  9. Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.

    记录一下遇到的问题. 最近在搭环境时出现了这个错误,网上找了许多方案都没法解决,说的大概都是cuda版本和pytorch版本不兼容,我按照pytorch官网的指令安装了许多次都没用. 我的cuda版本 ...

最新文章

  1. HDU - 5874 Friends and Enemies 完全二分图
  2. YOLO在升级 | PP-YOLO v2开源致敬YOLOV4携带Tricks又准又快地归来（附论文与源码）...
  3. Springboot中给图片添加文字水印
  4. 【Python 2.x和Python 3.x版本有什么区别】
  5. SpringMVC组件说明以及配置代码
  6. 3. Longest Substring Without Repeating Characters无重复字符的最长子串
  7. 使用showMessageDialog显示消息框
  8. java 工程文件改名后显示绿色加号_Gradle学习记录018 构建java JVM工程 part1
  9. PHP读取TXT中文乱码的解决方法
  10. mysql索引与事务笔记_《MySQL技术内幕：InnoDB存储引擎》读书笔记五-锁、索引及事务...
  11. 基于WCF的Silverlight 发布的注意事项
  12. 假设某台台式计算机的内存储器容量为128,假设某台式计算机的内存储器容量为128MB，硬盘容量为10GB。硬盘的容量是内存容量的60倍。...
  13. 国家统计局 2019年省市区数据（自取）
  14. 商品秒杀系统设计思路
  15. php页面强制横屏,Css实现手机端页面强制横屏(仅适用与一屏页面)
  16. 计数器函数 php,PHP用函数嵌入网站访问量计数器
  17. 设计模式-工厂模式（学习）
  18. 智慧园区数据可视化大屏的优势分析
  19. 蘑菇街Android组件与插件化
  20. matlab找多项式最值,如何利用matlab计算最小多项式

热门文章

  1. 基于图像的光照（Image-Based Lighting, IBL）概述
  2. C++实现 数字游戏之拼出最大数
  3. 戴尔服务器能用大白菜装系统吗,戴尔如何用u盘装系统xp
  4. MySQL怎么用x28改变颜色_MySQL SELECT练习题*28
  5. 稳健收益，缺你不可—A股优秀的基金和基金经理
  6. 北京大学肖臻老师《区块链技术与应用》公开课笔记23——ETH挖矿难度调整篇
  7. 求n以内的最大素数 ← C++
  8. 浏览器使用华视电子设备读取身份证信息
  9. 【web-攻击用户】（9.4）跨域捕获数据——通过注入HTML捕获数据、注入CSS捕获数据、JavaScript劫持
  10. 原生JS 和 jQuery 通过url传递 和 接收 自定义参数