code: def assert_torch_info(torch_info):
      install_torch_version = torch_info["version"]
      install_cuda_version = torch_info["cuda_version"]
  
      current_cuda_version = ".".join(torch.version.cuda.split(".")[:2])
      current_torch_version = ".".join(torch.__version__.split(".")[:2])
  
      if (
          install_cuda_version != current_cuda_version
          or install_torch_version != current_torch_version
      ):
          raise RuntimeError(
              "PyTorch and CUDA version mismatch! DeepSpeed ops were compiled and installed "
              "with a different version than what is being used at runtime. Please re-install "
              f"DeepSpeed or switch torch versions. DeepSpeed install versions: "
              f"torch={install_torch_version}, cuda={install_cuda_version}, runtime versions:"
              f"torch={current_torch_version}, cuda={current_cuda_version}"
          )
  
imports: import os
  import time
  import torch
  import importlib
  from pathlib import Path
  import subprocess
  from abc import ABC, abstractmethod
label_desc: Use image node_id 4 for calling a global function with example usage: assert_torch_info(torch_info) without return types