code: def _get_nvml_gpu_handle(
      device: Union[ivy.Device, ivy.NativeDevice], /
  ) -> int:
      global dev_handles
      if device in dev_handles:
          return dev_handles[device]
      gpu_idx = int(device.split(":")[-1])
      handle = pynvml.nvmlDeviceGetHandleByIndex(gpu_idx)
      dev_handles[device] = handle
      return handle
  
imports: import os
  import gc
  import abc
  import math
  import psutil
  import warnings
  import types
  from typing import Type, Optional, Tuple
  from typing import Union, Callable, Iterable, Any
  import ivy
  from ivy.func_wrapper import handle_out_argument, to_native_arrays_and_back, inputs_to_native_arrays, handle_nestable, handle_array_like_without_promotion, handle_backend_invalid
  from ivy.utils.exceptions import handle_exceptions
label_desc: Use image node_id 2 for calling a global function with example usage: _get_nvml_gpu_handle() and returns: handle, dev_handles