code: def _get_gpu_name():
      try:
          import pynvml
      except (ImportError, ModuleNotFoundError):
          return None
  
      pynvml.nvmlInit()
      handle = pynvml.nvmlDeviceGetHandleByIndex(0)
      cuda_capability, _ = pynvml.nvmlDeviceGetCudaComputeCapability(
          handle
      )
      pynvml.nvmlShutdown()
      if cuda_capability == 8:
          return "a100"
      elif cuda_capability == 9:
          return "h100"
      else:
          return None
  
file_path: NeMo/nemo/core/config/hydra_runner.py
imports: import argparse
  import functools
  import os
  import sys
  from typing import Any, Callable, Optional
  from hydra._internal.utils import _run_hydra, get_args_parser
  from hydra.core.config_store import ConfigStore
  from hydra.types import TaskFunction
  from omegaconf import DictConfig, OmegaConf
label_desc: Use image node_id 1 for calling a global function with example usage: _get_gpu_name() and returns: str, None, str, None