code: def _parse_colossalai_version():
      """
      Get the Colossal-AI version information.
  
      Returns:
          colossalai_version: Colossal-AI version.
          torch_version_for_aot_build: PyTorch version used for AOT compilation of CUDA kernels.
          cuda_version_for_aot_build: CUDA version used for AOT compilation of CUDA kernels.
      """
      # colossalai version can be in two formats
      # 1. X.X.X+torchX.XXcuXX.X (when colossalai is installed with CUDA extensions)
      # 2. X.X.X (when colossalai is not installed with CUDA extensions)
      # where X represents an integer.
      colossalai_version = colossalai.__version__.split("+")[0]
  
      try:
          torch_version_for_aot_build = colossalai.__version__.split(
              "torch"
          )[1].split("cu")[0]
          cuda_version_for_aot_build = colossalai.__version__.split(
              "cu"
          )[1]
      except:
          torch_version_for_aot_build = None
          cuda_version_for_aot_build = None
      return (
          colossalai_version,
          torch_version_for_aot_build,
          cuda_version_for_aot_build,
      )
  
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 4 for calling a global function with example usage: _parse_colossalai_version() and returns: colossalai_version,  torch_version_for_aot_build,  cuda_version_for_aot_build