pytorch模型加载失败报Attempting to deserialize object on a CUDA device but torch.cuda.is_available()...

环境:

X64平台: ubuntu18.04 torch版本:1.8.0
ARM平台: linux torch版本:1.8.0

描述:

在ubuntu X64平台使用GPU训练的模型,放到ARM平台使用CPU加载模型时报如下错误:
pytorch模型加载失败报Attempting to deserialize object on a CUDA device but torch.cuda.is_available()..._第1张图片

解决方法:

torch.load括号后面添加map_location=torch.device(‘cpu’)或者map_location=‘cpu’

在这里插入图片描述

你可能感兴趣的:(问题相关,pytorch,深度学习,人工智能)