1、修改torch.load,添加选择cpu
torch.load(opt.model, map_location='cpu')
2、将cuda()修改,首先用device判断一下torch是否可用,不行就用cpu,将原来直接写的.cuda()写成如下:
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
.cuda() -->(修改为) .to(device)
1、修改torch.load,添加选择cpu
torch.load(opt.model, map_location='cpu')
2、将cuda()修改,首先用device判断一下torch是否可用,不行就用cpu,将原来直接写的.cuda()写成如下:
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
.cuda() -->(修改为) .to(device)