明明之前代码好好的
model = torch.nn.DataParallel(model, device_ids=[0,1]).cuda()
原来是代码开始有个
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
明明之前代码好好的
model = torch.nn.DataParallel(model, device_ids=[0,1]).cuda()
原来是代码开始有个
os.environ["CUDA_VISIBLE_DEVICES"] = '0'