方法1
nvidia-smi
显示窗口
方法1升级版持续输出GPU占用率
但是有时我们希望不仅知道那一固定时刻的GPU使用情况,我们希望一直掌握其动向,此时我们就希望周期性地输出,比如每 10s 就更新显示。 这时候就需要用到 watch命令,来周期性地执行nvidia-smi命令了。
了解一下watch的功能:
$ whatis watch
watch(1) - execute a program periodically, showing output fullscreen12
作用:周期性执行某一命令,并将输出显示。
watch的基本用法是:
$ watch [options] command1
最常用的参数是 -n, 后面指定是每多少秒来执行一次命令。
监视显存:我们设置为每 10s 显示一次显存的情况:
$ watch -n 10 nvidia-smi1
显示如下:
这样,只要开着这个命令行窗口,就可以每十秒刷新一次,是不是很方便呢?
如果我们希望来周期性地执行其他命令行操作,那么就可以简单地更换后面的nvidia-smi即可,So Cool !
方法2
pip install gpustat
gpustat -cp
查看显卡硬件型号
ubuntu-drivers devices
从上图可以看出,我的显卡是:GM107M [GeForce GTX 860M]
,推荐安装的版本号是:nvidia-driver-390 - distro non-free recommended
。