pyotrch学习笔记——模型与数据放到GPU上

1. 将模型或数据放在GPU上

数据：

data = data.cuda()

模型：

model.cuda()

可以看出数据是返回值，模型是自身转换。

2. device查看

查看数据是否在GPU上：

tensor.device

查看模型是否在GPU上：

next(model.parameters()).device