监测linux下英伟达显存使用情况，输出nvidia-smi的部分结果

# -*- coding: utf-8 -*-
"""
用于监测linux下英伟达显存使用情况，输出nvidia-smi的部分结果，保存到同目录下一txt文件内
keyword后边列表内为筛选需要显示行中的关键词的
time.sleep(60)括号内数值表示查询间隔，单位秒

Created on Thu Jul 15 16:00:39 2021

@author: haijiao
"""

import os
import time


start_time = time.strftime("%m_%d_%H_%M")
keyword = ['./AdasApp', '7981MiB', '7982MiB', '12052MiB']
while 1:
    p = os.popen('nvidia-smi')
    nvidia = p.read()
    memory = nvidia.split('\n')
    with open(f'nvidia_{start_time}.txt', 'a') as f:
        for lines in memory:
            if any(word in lines for word in keyword):
                f.write(time.strftime("%Y-%m-%d %H:%M:%S "))
                f.write(lines)
                f.write('\r\n')
        f.write('\r\n')
    time.sleep(60)