code: def _check_torch_version():
      """
      Get the PyTorch version information.
  
      Returns:
          torch_version: PyTorch version.
          torch_cuda_version: CUDA version required by PyTorch.
      """
      # get torch version
      # torch version can be of two formats
      # - 1.13.1+cu113
      # - 1.13.1.devxxx
      torch_version = torch.__version__.split("+")[0]
      torch_version = ".".join(torch_version.split(".")[:3])
  
      # get cuda version in pytorch build
      try:
          torch_cuda_major = torch.version.cuda.split(".")[0]
          torch_cuda_minor = torch.version.cuda.split(".")[1]
          torch_cuda_version = f"{torch_cuda_major}.{torch_cuda_minor}"
      except:
          torch_cuda_version = None
  
      return torch_version, torch_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 6 for calling a global function with example usage: _check_torch_version() and returns: torch_version,  torch_cuda_version