python获取gpu的进程信息

在Python中，你可以使用`psutil`库来获取GPU进程的信息。首先，需要安装这个库，如果你还没有安装，可以使用pip命令：

pip install psutil

然后，你可以通过以下步骤来获取GPU相关进程的信息：

1. 导入所需的模块：

import os
from psutil import Process, NoSuchProcess

2. 获取当前运行的所有进程，包括GPU进程：

def get_gpu_processes():
    try:
        # 在Linux系统上，GPU进程通常关联有nvidia-smi命令
        nvidia_smi_cmd = "pgrep -f 'nvidia-smi'"
        pids = [int(pid) for pid in os.popen(nvidia_smi_cmd).read().split()]
        
        processes = []
        for pid in pids:
            try:
                process = Process(pid)
                if 'nvidia' in str(process.open_files()).lower():
                    processes.append((process, pid))
            except NoSuchProcess:
                pass
    except Exception as e:
        print(f"Failed to get GPU processes: {e}")
        processes = []

    return processes

3. 这段代码会返回一个包含GPU进程的元组列表，每个元素包含一个`psutil.Process`对象以及进程ID。例如：

gpu_processes = get_gpu_processes()
for process, pid in gpu_processes:
    print(f"PID: {pid}, Command: {process.cmdline()}")