code: def _check_cuda_version():
      """
      Get the CUDA version information.
  
      Returns:
          cuda_version: CUDA version found on the system.
      """
  
      # get cuda version
      if CUDA_HOME is None:
          cuda_version = CUDA_HOME
      else:
          try:
              raw_output = subprocess.check_output(
                  [CUDA_HOME + "/bin/nvcc", "-V"],
                  universal_newlines=True,
              )
              output = raw_output.split()
              release_idx = output.index("release") + 1
              release = output[release_idx].split(".")
              bare_metal_major = release[0]
              bare_metal_minor = release[1][0]
              cuda_version = f"{bare_metal_major}.{bare_metal_minor}"
          except:
              cuda_version = None
      return cuda_version
  
file_path: colossalai/colossalai/cli/check/check_installation.py
imports: import subprocess
  import click
  import torch
  from torch.utils.cpp_extension import CUDA_HOME
  import colossalai
label_desc: Use image node_id 7 for calling a global function with example usage: _check_cuda_version() and returns: cuda_version