win10 實現nvidia-smi 刷新顯示

在Linux系統下,我們可以用watch 命令刷新顯示 nvidia-smi 的信息。

在win系統下,我們用python腳本也實現了類似的功能。代碼如下:

 

import time
import subprocess
import locale
import codecs
import os
cmd = "nvidia-smi"
interval = 3
while True:
    ps = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, shell=True)
    for i in range(0,13):
        data = ps.stdout.readline()
        # bytes轉化爲str
        data = str(data)
        # 判斷是否開始於"b'",去掉
        if data.startswith('b\''):
            data = data[2:]
        #同樣去掉尾部
        if data.endswith('\\r\\n\''):
            data = data[:len(data)-5]
        print(data)
    print("\n\n\n\n\n\n\n")
    time.sleep(interval)      

保存爲.py 文件運行即可。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章