code: def test_gpu_setup():
      is_gpu_instance = utils.is_gpu_instance()
      if is_gpu_instance:
          assert (
              torch.cuda.is_available()
          ), "## Ohh its NOT running on GPU !"
  
file_path: serve/ts_scripts/sanity_utils.py
imports: import glob
  import json
  import os
  import subprocess
  import sys
  from pathlib import Path
  import torch
  from ts_scripts import marsgen
  from ts_scripts import tsutils
  from ts_scripts import utils
  from ts_scripts.tsutils import generate_grpc_client_stubs
label_desc: Use image node_id 4 for calling a global function with example usage: test_gpu_setup() without return types