python - 如何使用关键字在 nvidia-smi 中使用 PID 杀死 GPU 上的进程？

Question

如何在终端中终止特定程序(例如 python)在 GPU 上运行的进程？例如，上图中有两个进程正在使用 python 运行，杀死它们即可在 nvidia-smi 中查看下图

Answer 1

接受的答案对我不起作用，可能是因为 nvidia-smi 在不同版本/硬件上有不同的格式。

我正在使用更简洁的命令:

nvidia-smi | grep 'python' | awk '{ print $3 }' | xargs -n1 kill -9

您可以替换 awk 表达式中的 $3 以适合您的 nvidia-smi 输出。它是 PID 出现的第 n 列。