问题描述
在自己创建的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’)