在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上，以加速計算并提高訓練性能。
