注意
本文方案仅适用于NVIDIA显卡,对于Intel显卡和AMD显卡本文暂不做讨论。毕竟NVIDIA的显卡才是训练神经网络的大众之选。
方案1
使用nvidia-smi命令,输出为:

如果想要不间断持续刷新GPU的状态,使用:
nvidia-smi -l seconds
nvidia-smi --loop=seconds
上述命令任选其一,注意把seconds换成你想刷新的时间间隔(秒)。命令运行的效果就是持续输出上图中的内容(注意不是在原有输出上进行刷新,而是持续输出新的信息)。按Ctrl+C停止刷新。
方案2
有人写了个小工具gpustat把nvidia-smi封装了起来,用起来很爽很方便,推荐给大家。
首先安装:
pip install gpustat
然后使用:
gpustat -cp
输出为:

注意-c选项为显示进程名,-p选项为显示进程PID。如果想要不间断持续输出,请使用:
gpustat -cp -i 1
输出为:

最低0.47元/天 解锁文章
12万+

被折叠的 条评论
为什么被折叠?



