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

最新推荐文章于 2025-02-10 18:45:26 发布

Xovee

最新推荐文章于 2025-02-10 18:45:26 发布

阅读量4.4k

点赞数 5

分类专栏：报错文章标签： PyTorch Torch CUDA device 报错加载模型

转载需经过作者同意：xovee@ieee.org

本文链接：https://blog.csdn.net/xovee/article/details/119874825

版权

报错专栏收录该内容

53 篇文章

订阅专栏

这篇博客介绍了在使用PyTorch 1.9加载模型时遇到的CUDA设备不匹配错误，提供了从原代码`model.load_state_dict(model_path)`到修复后的`torch.load(map_location='cuda:0')`的解决方案。确保了在正确设备上运行模型加载操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

报错

File "/home/xovee/anaconda3/envs/pytorch/lib/python3.9/site-packages/torch/serialization.py", line 142, in validate_cuda_device
    raise RuntimeError('Attempting to deserialize object on CUDA device '
RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1. Please use torch.load with map_location to map your storages to an existing device.

环境

PyTorch 1.9

出错原因

显卡设备不匹配，原为'cuda:1'，现为'cuda:0'。

解决方案

原代码：

model.load_state_dict(model_path)

现代码：

loaded_state = torch.load(model_path, map_location='cuda:0')
model.load_state_dict(loaded_state)

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

Xovee

关注关注

5
点赞
踩
5

收藏

觉得还不错? 一键收藏
2
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
打赏
打赏
打赏举报

举报

专栏目录

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

weixin_39417324的博客

05-12

2253

根本原因：torch和cuda或者是cudnn版本不匹配详细教程可以参考我前边的文章 https://blog.csdn.net/weixin_39417324/article/details/122090285

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

04-17

379

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

2 条评论您还未登录，请先登录后发表或查看评论

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

最新发布

m0_52459066的博客

02-10

377

根据服务器的CUDA版本选取合适的pytorch对应的CUDA版本，比如，我的服务器CUDA版本为11.3，所以我也要找pytorch对应的CUDA版本，然后再看看哪个pytorch版本适合你的项目。然后安装相应的版本即可。有GPU版的错误原因：python中安装的pytorch对应的CUDA版本和服务器上的显卡驱动对应的CUDA版本不兼容，显卡驱动对应的CUDA版本应当>=pytorch对应的CUDA版本。

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

“我尊重同性恋，我理解抑郁症，我会喂楼底下的流浪狗，会给老奶奶让座位，会对店员说谢谢......纵然，我平庸，长相普通，脾气不好，曾经悲伤难过，望而不得，生活的苦也在吃，但我依然，想努力的成为一个温柔善良的人。”

10-29

7969

RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count1 项目场景2 问题描述3 原因分析4 解决方案 1 项目场景使用LSTM进行交叉路口的车辆行为识别。 2 问题描述在使用PyTorch加载模型时出现错误语句： RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.de

解决问题：RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available()

qq_42479987的博客

04-13

3564

问题： Traceback (most recent call last): File "test.py", line 22, in <module> model = loadmodel() File "/home/joshuayun/Desktop/IBD/loader/model_loader.py", line 48, in loadmodel checkpoint = torch.load(settings.MODEL_FILE) File "/home/jo

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

weixin_43442053的博客

03-16

2654

本程序所支持的环境可以用CPU版本的pytorch架构即可 RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location='cpu' to map your storages to the CPU.

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

hkx11111的博客

08-06

2352

yolov7调试过程CUDA报错

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

Anne332的博客

07-15

1283

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device(‘cpu’) to map your storages to the CPU.解决办法首先找到出错语句我这里是

【遇到问题-已解决】RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.

计算机视觉

01-30

1205

问题原因：采用的gpu是本机没有的，比如我的机器1个GPU但是我设置的是使用9卡就会导致这个问题。解决办法：这是因为gpu是'9'导致的，改成0就行。

Bug解决：RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available()

据说大佬都喜欢黑色！

09-18

7647

遇到一个错误： RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=‘cpu’ ...

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

热门推荐

入门记

09-23

1万+

意思：运行时错误：正在尝试反序列化CUDA设备上的对象，但torch.CUDA.is_available（）为False。如果您在只有CPU的机器上运行，请使用torch.load和map_location='pu'将您的存储映射到CPU。：torc.CUDA.is_available（）为False。pip list 查看torch版本，查看匹配的显卡驱动，显示正常。卸载torch，重新安装了包含cuda包的torch版本。软件：torch，显卡驱动，CUDA。查看CUDA版本，没有CUDA。

pytorch RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available

xingleitinghai的博客

07-08

3548

报错： RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device(‘cpu’) to map your storages to the CPU分析原因： 1、在加载mode

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

09-14

当你遇到这个错误时，意味着你的代码试图在一个CUDA设备上反序列化对象，但是torch.cuda.is_available()返回了False。这通常是因为你的机器没有可用的GPU，或者你的代码没有正确配置GPU。为了解决这个问题，你可以使用torch.load函数的map_location参数将存储器映射到CPU上。具体来说，你可以将map_location参数设置为torch.device('cpu')，以便在CPU上加载和处理模型。另外，如果你遇到了RuntimeError: CUDA out of memory错误，这意味着你的GPU显存不足。解决这个问题的方法有几种： 1. 将batch_size减小。减小每个批次中样本的数量可以减少显存的使用量。 2. 使用item()属性来获取torch变量的标量值。这可以释放不再需要的GPU显存。 3. 在测试阶段添加如下代码，以确保在每个批次之间释放GPU显存： ``` with torch.no_grad(): # your code here ``` 通过采取这些措施，你应该能够解决RuntimeError和CUDA out of memory错误。