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