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

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

效果图

效果

代码总览

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之外,其他命令套用即可。

版权声明:
作者:小筱筱
链接:https://jkboy.com/archives/10123.html
来源:随风的博客
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
打赏
海报
win上实现linux上watch功能,基于pyqt5在windows下实时刷新nvidia-smi,可套用其他类似命令!
在linux中我们可以使用watch和nvidia-smi来不停刷新GPU信息,但是在windows上没有watch,只能通过nvidia-smi -l 1来实现最快一秒一刷新,但是显示的效果是一……
<<上一篇
下一篇>>