成功解决RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be

错误代码

问题原因：这是由于Input type 与weight type的类型不符合，可以注意到Input type是没有cuda的，而weight type是在cuda上也就是有gpu的加载的。（错误代码后的or可以不用管）

解决办法：既然是类型不符导致的错误，那么就将类型转为一致，代码详见下面：

1.
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
input=input.to(device)
2.
input=input.cuda()#这两种代码任选其一即可