不修改pytroch的基本代码，只需要作如下修改即可：

为模型配置cuda：

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model.to(device) # 这里的model是实例化的模型，及 model = xxxNet()

将数据传入gpu：
在循环训练代码时，原来是：

inputs = torch.from_numpy(x_train)
labels = torch.from_numpy(y_train)

改为：

inputs = torch.from_numpy(x_train).to(device)
labels = torch.from_numpy(y_train).to(device)