pytorch版本問題：AttributeError: 'module' object has no attribute '_rebuild_tensor_v2'

用pytorch加載訓練好的模型的時候遇到了以下的問題：

AttributeError: 'module' object has no attribute '_rebuild_tensor_v2'

 

到網上查了一下是因爲訓練模型時使用的是新版本的pytorch，而加載時使用的是舊版本的pytorch。

解決方法：

一、既然是pytorch版本較老，那最簡單的解決方法固然是簡單的升級一下pytorch就ok了。

二、國外的大神給了另外一種解決方法，就是在程序開頭添加下面的代碼，便可以使老版本pytorch兼容新版本pytorch，參考連接https://discuss.pytorch.org/t/question-about-rebuild-tensor-v2/14560

 1 import torch._utils  2 try:  3  torch._utils._rebuild_tensor_v2  4 except AttributeError:  5     def _rebuild_tensor_v2(storage, storage_offset, size, stride, requires_grad, backward_hooks):  6         tensor = torch._utils._rebuild_tensor(storage, storage_offset, size, stride)  7         tensor.requires_grad = requires_grad  8         tensor._backward_hooks = backward_hooks  9         return tensor 10     torch._utils._rebuild_tensor_v2 = _rebuild_tensor_v2