加载模型出现 RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available()

最新推荐文章于 2024-03-11 20:31:01 发布
最新推荐文章于 2024-03-11 20:31:01 发布
阅读量6.2k 收藏 9
点赞数 3
分类专栏： 深度学习 python pytorch 文章标签： 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/binbinczsohu/article/details/107944257
版权
python 同时被 3 个专栏收录
26 篇文章 2 订阅
订阅专栏
pytorch
12 篇文章 0 订阅
订阅专栏
深度学习
6 篇文章 1 订阅
订阅专栏

 

今天在加载模型的时候出现了这个报错

 

问题现象

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.

 

问题分析

根据报错的提示，这个问题应该是因为模型是在GPU上训练的，然后我是在CPU上加载两者环境差异导致的，如果朋友们的电脑有CPU，那么可以参考我下面的方法

 

解决办法

在原来的加载方法中指定参数map_location为cpu


checkpoint = torch.load("ckpts/cdnn/model.tar")

改为

checkpoint = torch.load("ckpts/cdnn/model.tar",map_location='cpu')

 

DanielFaster
关注
  • 3
    点赞
  • 踩
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
torch.cuda.is_available()返回false的解决办法
qianchess的博客
12-25 9万+
在使用pytorch的时候，有时会遇到torch.cuda.is_available()返回false的问题，一般出现这个的原因是因为驱动没有安装好的缘故。解决的方案是： 1、安装nvidia相应的驱动版本和对应的cuda和cudnn。 sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update sudo apt inst...
libcudnn8-dev_8.0.2.39-1+cuda10.1_amd64_ubuntu18.04.zip
08-17
cudnn_v8.0.2.39 for ubuntu18.04 cuda10.1,安装cudnn发现需要去nvidia官网注册后下载，由于下载缓慢，这里提供一个mirror，for cuda10.1 ubuntu18.04，包含cuDNN Runtime Library for Ubuntu18.04 (Deb)和cuDNN Developer Library for Ubuntu18.04 (Deb)
torch.cuda.is_available()=False
最新发布
qq_50988206的博客
03-11 584
2、在pycharm终端输入 conda create -n pytorch 创建一个新的名为pytorch的虚拟环境。1、删除原来的虚拟环境，假如原虚拟环境为pytorch-old，输入以下命令，先退出当前虚拟环境，然后删除原虚拟环境。问题： 显示torch.cuda.is_available()=False。解决办法：说明这个虚拟环境不可用，删除虚拟环境，重建一个新的虚拟环境。）上找到符合自己的安装命令，以下是我的安装命令。3、在pytorch官网（
Python RuntimeError: thread.__init__() not called解决方法
09-22
主要介绍了Python RuntimeError: thread.__init__() not called解决方法,需要的朋友可以参考下
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码： with torch.no_grad(): # 停止自动反向计算梯度 参考： https://wisdomai.xyz/tool/pytorch/archives/2301 https://ptorch.com/news/160.html 作者：菜叶儿掉啦
RuntimeError: Cannot run the event loop while another loop is running(目前没有解决)
12-21
代码如下: import tornado.ioloop import tornado.web from tornado.httpclient import HTTPClient, AsyncHTTPClient class MainHandler(tornado.web.RequestHandler): # 同步 def get(self): h_c = HTTPClient() res = h_c.fetch("http://www.baidu.com") # print(res) # pass
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() i
入门记
09-23 7128
意思：运行时错误：正在尝试反序列化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。
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is Fal
qq_44936246的博客
08-16 1万+
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.
pytorch出错: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False
热门推荐
六五酥的博客
05-10 3万+
错误一：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. 解决方法：在torch.l
CUDA_Runtime_API.pdf
08-26
CUDA原生态变成API文档，内含所有cuda最新版本编程需要的接口。
人脸检测卷积神经网络_卷积神经网络和计算机视觉的人脸面具检测算法
weixin_26704853的博客
09-26 2259
人脸检测卷积神经网络In this article, we explore an application of Computer Vision that is largely relevant to the global health crisis that is the Coronavirus. As many countries continue their desperate fight t...
Error attempting to get column ‘HEADIMAGE‘ from result set. Cause: java.sql.SQLException: 不支持的字符集
喵酱
08-28 282
## Error querying database. Cause: org.apache.ibatis.executor.result.ResultMapException: Error attempting to get column 'HEADIMAGE' from result set. Cause: java.sql.SQLException: 不支持的字符集 (在类路径中添加 orai18n.jar): ZHS16GBK。
opencv 与dlib 结合实现人脸融合
weixin_NineDays66
11-12 7525
融合流程检测面部标记旋转、缩放和转换第二张图像，使之与第一张图像相适应调整第二张图像的色彩平衡，使之与第一个相匹配把第二张图像的特性混合在第一张图像中实现流程找到人脸矩阵使用dlib提取面部标记用Dlib实现了论文One Millisecond Face Alignment with an Ensemble of Regression Trees中的算法 （ http://www.csc.kth...
pytorch报错：Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False
lyly223的博客
04-23 69
已经按照修改，依然报错，请问怎么办？
面部合成opencv+Dlib
BBZZ2的专栏
01-04 2301
主要步骤 面部合成的原理就是利用给定的两张图片I1和 I2生成 N张从I1渐变到I2的过度图片 Mi,i=1,…,N。这N过度图片生成的原理是一样的，通过一个参数α来控制混合的程度。  Mi=(1-α)I1+αI2                     （1） 当α接近0时，Mi,看起来比较像I1 ，当I2接近1时，Mi看起来比较像I2。当然啦，这个公式只是一个大概的意思。具体来说一共
pytorch RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available
xingleitinghai的博客
07-08 3154
报错： 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 a CUDA device but torch.cuda.is_available() is False
09-23
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False 这个错误通常发生在使用PyTorch加载保存在GPU上的模型时，但是当前环境没有可用的CUDA设备。要解决这个问题，你可以使用torch.load函数，并指定map_location参数为torch.device('cpu')或'map_location='cpu''，将模型加载到CPU上。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值