如何在CPU代码中更改此GPU代码？

Question

所有

这是我在PyTorch中的简单GPU代码。

如何在CPU代码中更改此内容？

.col-md-12{
padding: 0;
}

提前致谢。

Answer 1

在调用.cuda()方法之前输入此代码

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')

，然后将每个.cuda()替换为.to(device) here

的很好答案