code: def environment_check_for_cuda_extension_build():
      if not TORCH_AVAILABLE:
          raise ModuleNotFoundError(
              "[extension] PyTorch is not found while CUDA_EXT=1. You need to install PyTorch first in order to build CUDA extensions"
          )
  
      if not CUDA_HOME:
          raise RuntimeError(
              "[extension] CUDA_HOME is not found while CUDA_EXT=1. You need to export CUDA_HOME environment variable or install CUDA Toolkit first in order to build CUDA extensions"
          )
  
      check_system_pytorch_cuda_match(CUDA_HOME)
      check_pytorch_version(
          MIN_PYTORCH_VERSION_MAJOR, MIN_PYTORCH_VERSION_MINOR
      )
      check_cuda_availability()
  
file_path: colossalai/setup.py
imports: import os
  import sys
  from datetime import datetime
  from typing import List
  from setuptools import find_packages, setup
  from op_builder.utils import check_cuda_availability, check_pytorch_version, check_system_pytorch_cuda_match, get_cuda_bare_metal_version, get_pytorch_version, set_cuda_arch_list
label_desc: Use image node_id 1 for calling a global function with example usage: environment_check_for_cuda_extension_build() without return types