用py获取显卡的占用率

这是什么

这是一个py 编写的程序，功能上面是用于获取 NVIDIA 显卡的占用率，并通过串口将其发送出去。同时，程序也会读取串口接收到的数据并显示在终端上，这样方便调试。

注意

因为我用的是N卡所以这个程序限制N卡使用，然后你如果要测试需要把 port 变量设置为正确的串口名称，波特率也要配合一下

import subprocess
import serial
import timedef get_gpu_usage():# 运行 `nvidia-smi` 并获取输出result = subprocess.run(['nvidia-smi', '--query-gpu=utilization.gpu', '--format=csv,noheader,nounits'], stdout=subprocess.PIPE, text=True)# 解析输出gpu_usage = int(result.stdout.strip())return gpu_usagedef send_usage_via_serial(ser, usage):# 创建要发送的字符串usage_str = f"GPU Usage: {usage}%\n"  # 创建以 "GPU Usage: xx%" 格式的字符串print(f"Sending: {usage_str.strip()}")  # 调试输出，去掉末尾的换行符ser.write(usage_str.encode())  # 编码为字节流发送def read_from_serial(ser):if ser.in_waiting > 0:return ser.read(ser.in_waiting).decode('utf-8').strip()return Noneif __name__ == "__main__":port = 'COM10'baudrate = 115200timeout = 1try:# 初始化串口连接ser = serial.Serial(port, baudrate=baudrate, timeout=timeout)print(f"串口 {port} 已连接。")while True:try:# 获取 GPU 占用百分比gpu_usage = get_gpu_usage()# 显示在终端print(f"GPU Usage: {gpu_usage}%")# 通过串口发送 GPU 占用百分比send_usage_via_serial(ser, gpu_usage)# 读取并显示串口收到的信息received_data = read_from_serial(ser)if received_data:print(f"Received from serial: {received_data}")# 等待 0.9 秒time.sleep(0.2)except Exception as e:print(f"发生错误: {e}")except KeyboardInterrupt:print("程序已终止。")except Exception as e:print(f"初始化串口时发生错误: {e}")finally:if ser.is_open:ser.close()  # 确保在程序结束时关闭串口连接print(f"串口 {port} 已关闭。")