torch.load(file) 默认是使用cuda GPU，要改为使用CPU需要添加参数map_location=torch.device(‘cpu’) 如下：

torch.load(file, map_location=torch.device('cpu'))

这样就是用CPU了