【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’)