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

Logo

更多推荐