RuntimeError: Expected object of backend CPU but got backend CUDA for argument #4 'mat1'

根据错误提示找到问题处在F.linear(input, self.weight, self.bias)这里,参数input是在cuda里面的,进入linear.py,在参数self.weight, self.bias后面的赋值语句后面加上.cuda(0),即torch.Tensor(…).cuda(0),问题解决。

你可能感兴趣的:(python,深度学习与计算机视觉,pytorch,cuda)