1. 将模型或数据放在GPU上 数据: data = data.cuda() 模型: model.cuda() 可以看出数据是返回值,模型是自身转换。 2. device查看 查看数据是否在GPU上: tensor.device 查看模型是否在GPU上: next(model.parameters()).device