如何查看你使用的pytorch是否为GPU版本

>>> import torch
>>> torch.cuda.is_available()
若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本

>>> import torch
>>> print(torch.cuda.get_device_name(0))
返回的GPU型号为你使用的GPU型号

直接查看GPU型号的方法
打开终端在终端中输入如下命令：

nvidia-smi
1
4 查看cuda版本的方法
nvcc -V
1
5 持续查看GPU的占有率
watch命令的作用：周期性执行某一命令，并将输出显示到终端
-n参数：每隔几秒更新一次，设为1可以几乎实时查看
————————————————
版权声明：本文为CSDN博主「亦岚君」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。