PyTorch中to()和cuda()的区别

小锋学长生活大爆炸

发布于 2022-03-29 13:39:47

1.4K0

没有区别。

早期(张量和模型都要)：

x = x.cuda()
model.cuda()

后来：

device = torch.device('cuda') if cuda_available else torch.device('cpu')
x = x.to(device)
model = model.to(device)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2022/01/13 ，如有侵权请联系 cloudcommunity@tencent.com 删除