torch.load を map_location=torch.device('cpu') とともに使用して、ストレージを CPU にマッピングしてください。

NoSuchKey

おすすめ

転載: blog.csdn.net/qq_39691492/article/details/121120909