python监测某进程的gpu占用情况

在Python中，你可以通过第三方库如`psutil`和`nvidia-smi`来监控特定进程的GPU占用情况。首先，你需要安装这两个库：

pip install psutil nvidia-smi

然后，可以结合这两个库使用，下面是一个简单的示例，展示如何获取某个进程的GPU使用信息：

import psutil
from subprocess import Popen, PIPE

def get_gpu_usage(process_name):
    # 获取所有运行的进程
    process = None
    for proc in psutil.process_iter(['pid', 'name']):
        if proc.info['name'] == process_name:
            process = proc
            break

    if not process:
        print(f"Process {process_name} not found.")
        return

    try:
        # 使用nvidia-smi命令行工具查询GPU信息
        gpu_info_cmd = ["nvidia-smi", "--query-gpu=utilization.gpu", "--format=csv"]
        p = Popen(gpu_info_cmd, stdout=PIPE)
        output, _ = p.communicate()

        # 解析输出，找到目标进程对应的GPU利用率
        lines = output.decode('utf-8').split('\n')
        utilization_lines = [line.split(',') for line in lines[2:]]  # 跳过前两行标题
        target_gpu_index = next((index for index, line in enumerate(utilization_lines) if process.info['cmdline'][0] in line), None)

        if target_gpu_index is not None:
            # 输出GPU利用率
            utilization_percentage = float(utilization_lines[target_gpu_index][0])
            print(f"{process_name}'s GPU usage: {utilization_percentage}%")
        else:
            print(f"No matching GPU found for process {process_name}.")
            
    except Exception as e:
        print(f"Error occurred while querying GPU usage: {str(e)}")

# 替换你想监控的进程名
get_gpu_usage("your_process_name")

记得将`"your_process_name"`替换为你想要监控的实际进程名称。