code: def __init__(self):
      self.peak_memory: int = 0
      self.device_index = int(
          os.environ["CUDA_VISIBLE_DEVICES"].split(",")[0]
      )
  
file_path: optimum/tests/benchmark/memory_tracker.py
imports: import os
  import subprocess
  from contextlib import contextmanager
  from multiprocessing import Pipe, Process
  from multiprocessing.connection import Connection
label_desc: Use image node_id 1 to create a new MemoryTracker object  with example: obj = MemoryTracker()