code: def gpu_info(session_config=None):
      is_fp16_capable = fp16_capable()
      num_devices = device_lib.list_local_devices(
          session_config=session_config
      )
      return {
          "n_gpus": num_devices,
          "fp16_inference": is_fp16_capable,
      }
  
imports: import re
  import logging
  from tensorflow.python.client import device_lib
  from tensorflow.python.framework import config
label_desc: Use image node_id 2 for calling a global function with example usage: gpu_info(session_config) and returns: dict