在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 文件運行即可。