code: def get_gpu_memory_map():
      """Get the current gpu usage."""
      result = subprocess.check_output(
          [
              "nvidia-smi",
              "--query-gpu=memory.used",
              "--format=csv,nounits,noheader",
          ],
          encoding="utf-8",
      )
      gpu_memory = np.array(
          [int(x) for x in result.strip().split("\n")]
      )
      return gpu_memory
  
file_path: GraphGym/graphgym/utils/device.py
imports: import logging
  import os
  import subprocess
  import numpy
  import torch
  from graphgym.config import cfg
label_desc: Use image node_id 1 for calling a global function with example usage: get_gpu_memory_map() and returns: gpu_memory