根据错误提示找到问题处在F.linear(input, self.weight, self.bias)这里,参数input是在cuda里面的,进入linear.py,在参数self.weight, self.bias后面的赋值语句后面加上.cuda(0),即torch.Tensor(…).cuda(0),问题解决。
RuntimeError: Expected object of backend CPU but got backend CUDA for argument #4 'mat1'
最新推荐文章于 2023-08-07 21:05:57 发布