一招教你检测你的gpu状态
·
在本地计算机跑深度学习模型时,很多朋友想了解代码运行时有没有用到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信息。

更多推荐



所有评论(0)