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

小锋学长生活大爆炸

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

1K0

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

没有区别。

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

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 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度