如何使用给定的 GPU 杀死所有进程？

Question

我使用 CUDA 工具包在我的 Nvidia GPU 上执行一些计算。如何杀死使用给定 GPU 的所有进程？（立即杀死，即无需手动键入后面的 PID kill -9。）

例如使用 GPU 2 杀死所有进程：

Answer 1

遵循 Unix 哲学，您有一个列出使用给定 GPU 的进程的工具，以及一个杀死进程的工具。使用 shell 结构和文本处理工具将它们组合起来。

例如，要终止使用 GPU 2 的所有进程，可以执行以下命令：

kill $(nvidia-smi | awk '$2=="Processes:" {p=1} p && $2 == 2 && $3 > 0 {print $3}')

或者

kill $(nvidia-smi -g 2 | awk '$2=="Processes:" {p=1} p && $3 > 0 {print $3}')