数据、函数和模型(对象）迁移到GPU上

在训练深度学习模型的时候我们经常需要把数据、函数和模型(对象）迁移到GPU上，其法方法都是一样的。

1、.cuda()和.cpu()

.cpu()函数将目标放到cpu上，如：

data.cpu()
func.cpu()
model.cpu()

.cuda()将数据、函数和模型放于GPU上。

• 指定某个GPU

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "2