win上实现linux上watch功能,基于pyqt5在windows下实时刷新nvidia-smi,可套用其他类似命令!
2023-06-13 09:13:53 时间
在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之外,其他命令套用即可。
相关文章
- 建立基于Linux的实时系统(基于linux的实时系统)
- 传输文件:从Linux到Windows(传文件到linux)
- Linux下快速修改文件所属用户(linux修改文件所属用户)
- Linux网络IO实时监控实践(linux网络io监控)
- 程序Linux实现Windows程序互通(让linux运行win)
- 服务器之间的比较比较Linux与Windows Server 2008的优势和劣势(linux和2008)
- 安装双系统:Linux在Windows之下(linux双系统下载)
- 在Windows 8上安装Linux:挑战性的引导(win8引导linux)
- 解决Linux下Vi乱码问题(vi乱码linux)
- 和树Linux分支树:扩展系统的多种可能性(linux的分支)
- 跨平台远程办公:Windows 远程桌面上的 Linux(windows远程桌面linux)
- Linux运维:日常护航保障(linux运维日常工作)
- 『Linux系统中关闭防火墙』(关闭linux的防火墙)
- 如何在Linux系统中安装Windows操作系统(linux安装win)
- 「Linux命令联想」——快速学习Linux命令的技巧(linux命令联想)
- Linux查看账户信息:简单快捷的步骤(linux查看账户信息)
- 运行Windows程序?试试Linux下Wine(linux下wine)
- Linux 命令:查找和删除指定文件(linux查找删除文件)
- Linux网口流量监控:了解实时传输状况(linux查看网口流量)
- 破解Linux服务器,体验网络抓包之旅(网络抓包 linux)
- 对比分析:Linux系统与Windows系统的异同(linux系统与windows系统的区别)
- Linux下执行SQL脚本:灵活、快捷、方便!(linux执行sql脚本)