code: def test_nvml():
      try:
          nvml.nvmlInit()
      except nvml.NVMLError as ne:
          try:
              import subprocess
  
              subprocess.check_output("nvidia-smi", shell=True)
              raise
          except:
              pytest.skip("No nvidia device")
  
      device_count = nvml.nvmlDeviceGetCount()
      for device_index in range(device_count):
          handle = nvml.nvmlDeviceGetHandleByIndex(device_index)
  
          name = nvml.nvmlDeviceGetName(handle)
          print(f"gpu_{device_index} name: {name.decode()}")
  
          utilization = nvml.nvmlDeviceGetUtilizationRates(handle)
          print(
              f"gpu_{device_index} GPU utilization: {utilization.gpu}%"
          )
  
          memory = nvml.nvmlDeviceGetMemoryInfo(handle)
          print(
              f"gpu_{device_index} total memory: {int(memory.total/1024**2)}MB"
          )
  
      nvml.nvmlShutdown()
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_nvml() without return types