查看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