在使用Pytorch加载模型时出现这个错误语句。
原因：原本的模型是用两个GPU训练的，而你的电脑只有一个，所以会出错。
解决：
model = torch.load(model_path)
改为：
model = torch.load(model_path, map_location='cuda:0')
如果是4块到两块：就把map_location改为：map_location={'cuda:1': 'cuda:0'}
其余情况以此类推。。。
RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1
猜你喜欢
转载自blog.csdn.net/iamjingong/article/details/85308600
今日推荐
周排行