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

Question

如何在终端中为特定程序（例如 python）杀死 GPU 上正在运行的进程？例如，两个进程在顶部图片中使用 python 运行并杀死它们以在 nvidia-smi 中查看底部图片

标签： pythongpunvidiakeywordpid

Answer 1

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

我正在使用一个更干净的命令：

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

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