code: def _collect_memory_info(run_info):
      try:
          # Note: psutil is not installed in the TensorFlow OSS tree.
          # It is installable via pip.
          import psutil  # pylint: disable=g-import-not-at-top
  
          vmem = psutil.virtual_memory()
          run_info["machine_config"]["memory_total"] = vmem.total
          run_info["machine_config"][
              "memory_available"
          ] = vmem.available
      except ImportError:
          tf.compat.v1.logging.warn(
              "'psutil' not imported. Memory info will not be logged."
          )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import contextlib
  import datetime
  import json
  import multiprocessing
  import numbers
  import os
  import threading
  import uuid
  from six.moves import _thread
  from absl import flags
  import tensorflow
  from tensorflow.python.client import device_lib
  from utils.logs import cloud_lib
label_desc: Use image node_id 10 for calling a global function with example usage: _collect_memory_info(run_info) without return types