pytorch使用GPU

检查是否有可用的 GPU: 使用 torch.cuda.is_available() 函数来检查是否有 GPU 支持 CUDA。

将模型移至 GPU: 使用 .to(device) 方法将模型移至 GPU，其中 device 是一个字符串，通常设置为 "cuda:0"（表示第一个 GPU）。

将数据移至 GPU: 与模型相同，您还需要确保输入数据和标签也在 GPU 上。在将数据送入模型之前，同样使用 .to(device) 方法。