Skip to main content
 首页 » 编程设计

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

2024年02月06日16lyhabc

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

请您参考如下方法:

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

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

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

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