1 最简单的方法
>>>import torch>>> torch.cuda.is_available()
若返回为True,则使用的是GPU版本的torch,若为False,则为CPU版本
2 查看GPU型号的方法
>>>import torch>>>print(torch.cuda.get_device_name(0))
返回的GPU型号为你使用的GPU型号
3 直接查看GPU型号的方法
打开终端在终端中输入如下命令:
nvidia-smi
4 查看cuda版本的方法
nvcc-V
5 持续查看GPU的占有率
watch命令的作用:周期性执行某一命令,并将输出显示到终端
-n参数:每隔几秒更新一次,设为1可以几乎实时查看
watch-n1 nvidia-smi
具体显示结果如下所示:重要的参数主要是温度、内存使用、GPU占有率,具体如下红框所示。