pytorh中to(device) 和.cuda()的区别

可以指定 cpu 或者GPU。

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model.to(device)
#如果是多GPU
if torch.cuda.device_count() > 1:
  model = nn.DataParallel(model，device_ids=[0,1,2])
model.to(device)
————————————————
版权声明：本文为CSDN博主「Golden-sun」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_43402775/article/details/109223794

只能指定 GPU。

#指定某个GPU
os.environ['CUDA_VISIBLE_DEVICE']='1'
model.cuda()
#如果是多GPU
os.environment['CUDA_VISIBLE_DEVICES'] = '0,1,2,3'
device_ids = [0,1,2,3]
net  = torch.nn. Dataparallel(net, device_ids =device_ids)
net = net.cuda()
————————————————
版权声明：本文为CSDN博主「Golden-sun」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_43402775/article/details/109223794