使用GPU跑pytorch程序

使用GPU跑pytorch程序

1. 判断GPU是否可用

	device=torch.device('cuda' if torch.cuda.is_available() else 'cpu')

2. 将模型放到GPU上
   在我们实体化网络模型时，直接在后面加一行代码就可以。

	net= Net ()
	net.to(device)

3. 把数据放到GPU上

	inputs,labels=data
	inputs, labels =  data[0].to(device), data[1].to(device)
	# 或者
	inputs, labels= inputs.to(device),labels.to(device)