在本地计算机跑深度学习模型时,很多朋友想了解代码运行时有没有用到gpu的算力,这时就很推荐gpustat这个python三方库。没安装之前通常都是使用nvidia-smi指令。

nvidia-smi监测方法

没有gpustat时,通常使用nvidia-smi指令来实现查看NVIDIA GPU 的实时状态。它是NVIDIA提供的命令行工具,属于 NVIDIA 驱动程序工具集的一部分,只要系统安装了 NVIDIA GPU 驱动,就可以使用它。

nvidia-smi

持续监测

nvidia-smi -l 1

在终端中输入此代码,终端持续反馈gpu显卡信息。-l 1 表明每隔1秒更新GPU信息。

使用Ctrl+C键,可以终止监测

gpustat监测方法

gpustat的安装

直接pip进行库的安装

pip install gpustat


 

gpustat的使用

在终端中输入如下指令,将会持续检测计算机的gpu使用状况

gpustat

这种方法返回的结果更简明易懂,很直观的看到GPU使用情况

持续监测

若想使用gpustat持续检测

可以使用

gpustat -i 1

其中 -i 1 表明间隔更新GPU信息。

更多推荐