shell监视gpu使用情况

一次 nvidia-smi 的输出形如：

Tue Aug  9 23:05:08 2022       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.44       Driver Version: 440.44       CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  TITAN Xp            Off  | 00000000:01:00.0 Off |                  N/A |
| 37%   58C    P2    75W / 250W |   8481MiB / 12195MiB |      2%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      2292      C   python                                      8467MiB |
+-----------------------------------------------------------------------------+

其中 8481MiB / 12195MiB 显示了已用显存、显存总量。现用 shell 命令计算可用显存大小：

nvidia-smi | \
	grep -E "[0-9]+MiB\s*/\s*[0-9]+MiB" | \
	awk '{print ($9" "$11)}' | \
	sed "s/\([0-9]\{1,\}\)MiB \([0-9]\{1,\}\)MiB/\1 \2/" | \
	awk '{print $2 - $1}'

其中：

• | 管道，将前一命令结果传给后一命令；\ 续行；
• grep -E "[0-9]+MiB\s*/\s*[0-9]+MiB" 筛出包含使用情况的那行输出，得：

  | 37%   58C    P2    75W / 250W |   8481MiB / 12195MiB |      2%      Default |
  

• awk '{print ($9" "$11)}' 筛出已用显存、显存总量两列核心数据（这里是该行的第 9 列和第 11 列），空格间开，得：

  8481MiB 12195MiB
  

• sed "s/\([0-9]\{1,\}\)MiB \([0-9]\{1,\}\)MiB/\1 \2/"去掉 MiB 后缀（括号括住匹配组，\1、\2 引用匹配到的字符串），得：

   8481 12195
  

• awk '{print $2 - $1}' 计算两列的差，算得可用显存，得：

  3714
  

这是单张卡的情况，如果有多张卡，可能 | 要配合 xarg 用？