python - 如何在 bash 中使用解析 nvidia-smi 输出并将解析结果用作其他命令的输入？

Question

这是我的 nvidia-smi 的输出命令，我想解析它的输出并杀死在 GPU 上运行的所有 python 进程。我已经知道如何使用我编写的以下脚本杀死所有 python 进程。

kall () {
    pgrep $1 | xargs kill -9
}

现在我想杀死所有使用 GPU 的进程(获取 PID 号)，所以我想解析 nvidia-smi 输出。

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 367.27                 Driver Version: 367.27                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 1070    Off  | 0000:01:00.0      On |                  N/A |
|  0%   46C    P2    40W / 166W |   4902MiB /  8112MiB |      1%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|    0       917    G   /usr/lib/xorg/Xorg                             209MiB |
|    0      1683    G   compiz                                         159MiB |
|    0      3780    G   ...UI2016Q2,UsePasswordSeparatedSigninFlow<P   130MiB |
|    0     25962    C   python                                        2199MiB |
|    0     26103    C   python                                        2201MiB |
+-----------------------------------------------------------------------------+

Answer 1

你可以从 nvidia-smi 得到你想要的输出直接使用

$ nvidia-smi --query-compute-apps=pid --format=csv,noheader
917
1683
3780
25962
26103

参见联机帮助页 nvidia-smi --help-query-compute-apps有关与在 gpu 上运行的应用程序相关的查询的更多信息。