- 博客(2)
- 收藏
- 关注
原创 Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu
device = torch.device('cuda') torch.rand(8,1).to(device) 原因在于torch函数生成的tensor需要转到GPU上
2022-03-29 09:55:55
1698
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人