有numpy和torch两种类型的数据
z = mean.clone().detach() + eps * torch.exp(logstd)
源代码这边报错了，修改如下
eps = eps.cuda() z = mean.cuda()+ eps * torch.exp(logstd).cuda()
就是把和numpy有关的，比如reshape什么的后面加上cuda()
#报错记录#RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
猜您在找
【深度学习】RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
张量与数组运算报错(Use Tensor.cpu() to copy the tensor to host memory first;RuntimeError: Expected all tensors to be on the same device)
#报错记录#RuntimeError: Expected tensor for 'out' to have the same device as tensor for argument #2 'mat1'; but device 1 does not equal 0 (while checking arguments for addmm)
【E-14-Pytorch】RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm
[报错]-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same
RuntimeError: Expected object of backend CUDA but got backend CPU for argument #4 'mat1'
RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for ar
RuntimeError: invalid argument 0: Tensors must have same number of dimensions: got 1 and 2 at /opt/conda/conda-
RuntimeError: CUDA error: invalid device ordinal
Pytoch的nn.CrossEntropyLoss 报错 RuntimeError: expected scalar type Long but found Float