PyTorch GPU 安装后，如何使用 GPU 运行代码？

是的，您需要在代码中指定使用 GPU，否则 PyTorch 会默认使用 CPU。可以使用如下代码将模型转移到 GPU 上：

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

或者使用下面的代码指定使用哪个 GPU：

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
model.to(device)

在训练过程中，您还需要将输入数据和目标数据转移到 GPU 上：

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

这样可以最大化 GPU 的性能，加速训练过程。