pytorch ,简单的一句 torch.load(),模型都有这么多坑

1、python版本不同，3不能打开2，torch.load加载模型报错

2、多GPU，单GPU，训练的GPU号不一样都报错

3、ModuleNotFoundError: No module named ‘models’，这个报错不知道什么原因
嗯，在训练文件夹中同一个目录下，运行测试都没问题，可是我在另一个文件下，运行它用绝对路径，也不能运行报错，实在不行了

4、之前可以运行加载模型后来又不可以，pytorch ,版本不同也很有可能运行不了模型，

5、torch.save和torch.load 调用的应该是pickle库，用pickle.load()要求python能够找到类的定义。

不想保留源码的话，可以把模型转化成TorchScript之后保存。 所以保存完整的 model param ， 的一个模型，需要调用的源码， torch.save和torch.load 调用的应该是pickle库，用pickle.load()要求python能够找到类的定义。

不想保留源码的话，可以把模型转化成TorchScript之后保存。