深度学习 如何查看你使用的pytorch是否为GPU版本

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 -n 1 nvidia-smi

具体显示结果如下所示：重要的参数主要是温度、内存使用、GPU占有率，具体如下红框所示。

推荐文章

 您阅读本篇文章共花了：