[Question] Can nvitop keep a log/record of GPU-Utilization and store in a CSV? about nvitop HOT 4 CLOSED

from nvitop import Device, GpuProcess

all_devices = Device.all()
all_gpu_processes = []
for device in all_devices:
    all_gpu_processes.extend(device.processes().values())

all_device_snapshots = [device.as_snapshot() for device in all_devices]
all_gpu_process_snapshots = [process.as_snapshot() for process in all_processes]

import datetime
import time

import pandas as pd

from nvitop import ResourceMetricCollector

collector = ResourceMetricCollector(root_pids={1}, interval=2.0)  # log all devices and all GPU processes
df = pd.DataFrame()

with collector(tag='resources'):
    for _ in range(12):
        # Do something
        time.sleep(5)

        metrics = collector.collect()
        df_metrics = pd.DataFrame.from_records(metrics, index=[len(df)])
        df = pd.concat([df, df_metrics], ignore_index=True)
        # Flush to CSV file ...

df.insert(0, 'time', df['resources/timestamp'].map(datetime.datetime.fromtimestamp))
df.to_csv('results.csv', index=False)