设定GPU的模块

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 

注意：仔细查看你的设备GPU的编号，我的设备禁用了集成显卡，仅留一个显卡使用。

将数据发送至GPU上

这部分很简单，将model、input、optimizer等数据后加*.to(device)*，例如：

model.to(device)

查看GPU使用情况

1. 使用任务管理器查看：
   打开任务管理器，在性能标签页查看
   

2. 使用命令行窗口查看：
   依次输入以下命令即可

cd C:\Program Files\NVIDIA Corporation\NVSMI
 
nvidia-smi

参考文献

1. https://blog.csdn.net/ViatorSun/article/details/107235732
2. https://blog.csdn.net/weixin_42731241/article/details/94881528
3. https://blog.csdn.net/weixin_42535423/article/details/109320965