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 2. Please use torch.load with map_location to map your storages to an existing device.
这个报错的原因是训练的时候用了编号为2的GPU,但是在你用pytorch load这个模型运行的时候使用的不是2号GPU。
解决方法很简单,在load的时候加一个map_location:
#model.load_state_dict(torch.load(model_path)) #会报错的写法
model.load_state_dict(torch.load(model_path,map_location={'cuda:2': 'cuda:0','cuda:1': 'cuda:0'})) #将对应的gpu编号映射到正确的gpu上
RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 2.相关推荐
- 【错误处理】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 ...
- RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1.
成功解决 RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() i ...
- 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 ...
- 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 ...
- 解决RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1
我是在服务器上训练的模型,然后下载到本地进行测试,报错如下: 问题应该是训练和测试的GPU不是同一个,需要将GPU改成本地的. 解决方法,在加载模型的语句上,增加 map_location='cuda ...
- PyTorch笔记: GPU上训练的模型加载到CPU/错误处理Attempting to deserialize object on a CUDA device but torch.cuda.is_a
我之前在GPU上训练了一个模型,同时把模型的参数记录在resnet18_ultimate.pkl上 在本地的CPU上,我想把参数加载,于是一开始我是这么写代码的: import torch impor ...
- 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 ...
- 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 ...
- Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.
记录一下遇到的问题. 最近在搭环境时出现了这个错误,网上找了许多方案都没法解决,说的大概都是cuda版本和pytorch版本不兼容,我按照pytorch官网的指令安装了许多次都没用. 我的cuda版本 ...
最新文章
- rabbitmq 查询版本_基于rabbitmq解决分布式事务
- 排除字段重复行的SQL
- 特征值和特征向量(Eigenvalues and Eigenvectors)
- java框架之Spring 核心框架体系结构
- 10款免费工具:敏捷开发运维(DevOps)的好帮手
- VTK:结构化网格之StructuredGrid
- python 文本框_重置Python中的文本框
- OOB与COM交互读写本地文件
- JS键盘字母相应的keyCode值
- 用@resource注解方式完成属性装配
- python可以做力扣的题吗_力扣题目练习一
- DEDE的简略标题标签的使用问题
- Js获取处理日期时间
- Matlab画圆且坐标轴图片等比例显示
- 交流有功功率、无功功率、视在功率计算
- 娃娃机共享扫码平台开发
- appstore上架所需截屏的制作方法
- B端产品运营:学习笔记
- 哈工程计算机系保研率,武汉理工VS哈工程,20保研率哈工程高,哪所值得考?学长精准分析...
- Hive 导出数据的五种方式