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

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

版权声明：本文内容来自第三方投稿或授权转载，原文地址：https://blog.51cto.com/xfxuezhang/5857187，作者：小锋学长生活大爆炸，版权归原作者所有。本网站转在其作品的目的在于传递更多信息，不拥有版权，亦不承担相应法律责任。如因作品内容、版权等问题需要同本网站联系，请发邮件至ctyunbbs@chinatelecom.cn沟通。