pytorch如何查看tensor和model在哪个GPU上

查看tensor所在的设备：

data = data.cuda()#将数据转移到gpu上

print(data.device)  # 输出：cuda:0

data = data.cpu()#将数据转移到cpu上

print(data.device)  # 输出：cpu

查看model所在的设备

model = model.cuda()#将模型转移到gpu上

print(next(model.parameters()).device)  # 输出：cuda:0

model = model.cpu()#将模型转移到cpu上

print(next(model.parameters()).device)  # 输出：cpu