为什么我的简单PyTorch网络在GPU设备上无法工作？

我按照教程构建了一个简单的网络，但遇到了以下错误：

RuntimeError: 期望类型为torch.cuda.FloatTensor的对象，但为参数#4 ‘mat1’找到了类型torch.FloatTensor

有什么帮助吗？谢谢！

回答：

TL;DR
这是修复方法

inputs = inputs.to(device)

为什么？！
torch.nn.Module.to() 和 torch.Tensor.to() 之间存在细微的差异：虽然 Module.to() 是一个原地操作符，但 Tensor.to() 不是。因此

net.to(device)

会改变 net 本身并将其移动到 device。另一方面

inputs.to(device)

不会改变 inputs，而是返回一个位于 device 上的 inputs 的副本。为了使用这个“在设备上”的副本，你需要将其赋值给一个变量，因此

inputs = inputs.to(device)

学技术