[Pytorch学习]在GPU上训练

主要分为以下：

1. 定义GPU

device=torch.device("cuda:0"if torch.cuda.is_available() else "cpu")

2. 将模型放到GPU中

def Model(...):
	....
	....

model=Model().to(device)

3. 把输入和标签放到GPU中

inputs, labels= inputs.to(device), labels.to(device)

4. 接下来就可以正常训练拉