code: def get_gpu_memory(gpu_dev_id):
      free_mem = ctypes.c_uint64(0)
      total_mem = ctypes.c_uint64(0)
      check_call(
          _LIB.MXGetGPUMemoryInformation64(
              gpu_dev_id,
              ctypes.byref(free_mem),
              ctypes.byref(total_mem),
          )
      )
      return free_mem.value, total_mem.value
  
imports: import ctypes
  import functools
  import inspect
  import threading
  import tempfile
  import platform
  from contextlib import contextmanager
  from struct import calcsize
  from .base import _LIB, check_call, c_str, py_str, numeric_types, integer_types, long, _MAX_VALUE_64_BIT_UNSIGNED_, _MAX_VALUE_64_BIT_SIGNED_, _MAX_VALUE_FLOAT32_REPRESENT_
label_desc: Use image node_id 2 for calling a global function with example usage: get_gpu_memory(gpu_dev_id) and returns: free_mem,  total_mem