PyTorch使用cpu与gpu之间模型相互加载调用

作者：你好赵伟 | 2024-07-19 12:29:17

踩

torch使用cpu

我们在使用pytorch的过程，经常会需要加载模型参数，但是pytorch当中，GPU和CPU模型下加载的参数的类型是不同的，不能互相直接调用，下面分情况进行操作说明。

torch.load(PATH, map_location="cpu")

模型在GPU1上保存，运行在GPU0和1上

torch.load(PATH, map_location={'cuda:1':'cuda:0'})

torch.load(PATH, map_location="cuda:0")

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/851498