在PyTorch中使用GPU的方法是通过将模型和数据移动到GPU上进行加速计算。以下是在PyTorch中使用GPU的基本步骤：
import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model.to(device)
inputs, labels = inputs.to(device), labels.to(device)
for inputs, labels in dataloader:
inputs, labels = inputs.to(device), labels.to(device)
# 运行模型和优化步骤
通过这些步骤，您可以将整个训练过程移动到GPU上，以加速计算并提高训练性能。
