pytorch報錯：AttributeError: 'module' object has no attribute '_rebuild_tensor_v2'

轉載自：

https://blog.csdn.net/qq_24305433/article/details/80844548

 

由於訓練模型時使用的是新版本的pytorch，而加載時使用的是舊版本的pytorch

解決方法：

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

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

import torch._utils
try:
    torch._utils._rebuild_tensor_v2
except AttributeError:
    def _rebuild_tensor_v2(storage, storage_offset, size, stride, requires_grad, backward_hooks):
        tensor = torch._utils._rebuild_tensor(storage, storage_offset, size, stride)
        tensor.requires_grad = requires_grad
        tensor._backward_hooks = backward_hooks
        return tensor
    torch._utils._rebuild_tensor_v2 = _rebuild_tensor_v2