注意
本文方案仅适用于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
输出为:
<