解决pytorch报错:AssertionError: Invalid device id的问题 在服务器上训练的网络放到本地台式机进行infer，结果出现报错：AssertionError:Invaliddeviceid仔 AssertionError