如何使用关键字在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输出。这是发生PIDs的第n列。

Answer 2

可以在英伟达SMI中捕获Python，然后将PID传递给杀死9命令，例如：

sudo kill -9 $( nvidia-smi | grep 'python' | sed -n 's/|\s*[0-9]\s([0-9])\s.*/\1/p' | sed '/^$/d')