基于pyqt5在windows下实时刷新nvidia-smi，类似于linux下的watch，可自定义其他命令（win上实现watch）

引子

在linux中我们可以使用watch和nvidia-smi来不停刷新GPU信息，但是在windows上没有watch，只能通过nvidia-smi -l 1来实现最快一秒一刷新，但是显示的效果是一闪一闪的，而且不支持低于1秒的刷新频率。我于是有了用pyqt来呈现结果的想法，并且可以支持毫秒级别的刷新

效果

代码总览

• 开源地址：Github

import time
import subprocess
import threading
import locale
import codecs
import os
import sys
if hasattr(sys, 'frozen'): # 解决不同环境下打包目录位置错误
    os.environ['PATH'] = sys._MEIPASS + ";" + os.environ['PATH']
from PyQt5.Qt import QWidget
from PyQt5.QtGui import QIcon
from PyQt5.Qt import QLabel
from PyQt5.Qt import QApplication


# nvidia-smi获取
def get_cuda_status():
    cmd = "nvidia-smi" # 自定义命令，这里以nvidi-smi为例
    ps = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, shell=True)
    res = ''
    while True:
        data = ps.stdout.readline()
        data = str(data)
        if data == "b''":
            break
        if data.startswith('b\''):
            data = data[2:]
        if data.endswith('\\r\\n\''):
            data = data[:len(data)-5]
        data = data.replace('\\\\', '\\')
        res += data + '\n'
    return res

# GPU型号获取
def get_gpu_name():
    cmd = "nvidia-smi -L"
    ps = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, shell=True)
    res = ''
    while True:
        data = ps.stdout.readline()
        data = str(data)
        if data == "b''":
            break
        if data.startswith('b\''):
            data = data[2:]
        if data.endswith('\\r\\n\''):
            data = data[:len(data)-5]
        data = data.replace('\\\\', '\\')
        res += data + '&'
    return res


class Window(QWidget):
    def __init__(self):
        super().__init__()
        self.setWindowTitle(f'Cuda Supervisor @{get_gpu_name()}')
        self.resize(888, 950)
        self.setStyleSheet('background-color: #2A2A2A; color: white') 
        self.geometry().center()

        # 状态提示
        self.ts_status = QLabel(self)
        self.ts_status.setText("None")
        self.ts_status.setStyleSheet('font-size: 22px; color: white')
        self.ts_status.adjustSize()
        self.ts_status.move(10, 10)

        self.cuda_status_thread = True
        self.th = threading.Thread(target=self.refresh_info, args=())
        self.th.start()

    # 关闭时结束刷新进程
    def closeEvent(self, event):
        self.cuda_status_thread = False

	# 刷新cuda
    def refresh_info(self):
        while self.cuda_status_thread:
            self.ts_status.setText(get_cuda_status())
            self.ts_status.adjustSize()  
            time.sleep(0.05) # 刷新频率


if __name__ == '__main__':
    app = QApplication(sys.argv)

    window = Window()
    window.show()

    sys.exit(app.exec_())

使用方法

安装python环境以及pyqt5，如果有需要打包可以安装一下pyinstaller

• 安装pyqt： pip install pyqt5
• 安装pyinstaller： pip install pyinstaller

除nvidia-smi之外，其他命令套用即可。