【bug】RuntiomeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available()

问题描述

在自己创建的cuda环境下运行行人检测算法时,出现process has died,向上看到错误为cuda引起的(下图红框处)
在这里插入图片描述
这里是因为当前的conda环境中装的pytorch是cpu版本的,而代码中使用了gpu。

解决方案

1、改用gpu版本的pytorch

在pytorch官网查找安装指令https://pytorch.org/
下拉找到安装指令(根据自己电脑选择)
在这里插入图片描述

2、无gpu,继续使用cpu版

找到上图绿色框中的代码,将其中的
torch.load()
改成:
torch.load(
, map_location=‘cpu’)
在这里插入图片描述
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值