Pytorch细节记录 获取tensor所在gpu device = data.device 将tensor转移到指定gpu上 a = torch.randn(3,4,device='cuda:0') # 创建gpu0上的tensor a = a.to('cuda:1') # 转移到gpu1上 a = a.cuda(1) # 转移到gpu1 b = torch.randn(3,4,device=a.device) # 创建一个和a在一个设备上的tensor