主要介绍了pytorch模型载入之gpu和cpu互转操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

问题描述 有时在加载已训练好的模型时，会出现 out of memory 的错误提示，但仔细检测使用的GPU卡并没有再用且内存也没有超出。 经查阅发现原来是训练模型时使用的GPU卡和加载时使用的GPU卡不一样导致的。个人感觉，因为pytorch的模型中是会记录有GPU信息的，所以有时使用不同的GPU加载时会报错。 解决方法 gpu之间的相互转换。即，将训练时的gpu卡转换为加载时的gpu卡。 torch.load(‘modelparameters.pth’, map_location={‘cud