win上实现linux上watch功能，基于pyqt5在windows下实时刷新nvidia-smi，可套用其他类似命令！

小筱筱 • 2023-01-02 • 云技术社区 • 288 阅读

在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_())