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

报错信息

在这里插入图片描述

报错原因

开始遇到这个问题是真的有点懵,因为这是到了最后一个test.py的文件了,但是确不能跑通。
查了很多博客,终于站在巨人的肩膀上,找到了原因!!!
原来的代码模型是使用的多个GPU进行训练的,而当我实现时,只使用了一个,所以报错。

原本代码:torch.load(self.cfg.snapshot)
修改代码:torch.load(self.cfg.snapshot,map_location = 'cpu' )

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值