Tensor 有三种属性: dtype, device,和layout
问题: 数显cuda和cpu混乱,无法互相计算
解决:
device = torch.device('cuda', if torch.cuda.is_device()else 'cpu' )
x = torch.rand(1,1,3,4)
x = x.data.to(device)
Tensor 有三种属性: dtype, device,和layout
问题: 数显cuda和cpu混乱,无法互相计算
解决:
device = torch.device('cuda', if torch.cuda.is_device()else 'cpu' )
x = torch.rand(1,1,3,4)
x = x.data.to(device)