python脚本批量杀死占用GPU的所有进程/所有正在cuda上运行的程序

import os
import time


def get_exec_out(sxcute_str):
    out_list = os.popen(sxcute_str).readlines()
    return out_list


if __name__ == '__main__':
    excute_str = 'nvidia-smi'
    out_list = get_exec_out(excute_str)
    # print(out_list)
    for oo in out_list:
        if oo.find('python') != -1:
            # split()函数默认可以按空格分割，并且把结果中的空字符串删除掉，留下有用信息
            proc_list = oo.split()
            pid = proc_list[2].strip()
            kill_str = 'kill -9 ' + pid
            print(kill_str)
            time.sleep(0.3)
            os.system(kill_str)