code: def _get_cuda_bare_metal_version(cuda_dir):
      raw_output = subprocess.check_output(
          [cuda_dir + "/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]
  
      return raw_output, bare_metal_major, bare_metal_minor
  
imports: import os
  import pathlib
  import subprocess
  from torch.utils import cpp_extension
label_desc: Use image node_id 2 for calling a global function with example usage: _get_cuda_bare_metal_version(cuda_dir) and returns: raw_output,  bare_metal_major,  bare_metal_minor