我是在服务器上训练的模型,然后下载到本地进行测试,报错如下: 问题应该是训练和测试的GPU不是同一个,需要将GPU改成本地的。 解决方法,在加载模型的语句上,增加 map_location='cuda:0'