code: def test_nvidia():
      # pip install py3nvml
      import py3nvml
      from py3nvml import py3nvml as nvml
  
      inspect(py3nvml.get_free_gpus())
  
      nvml.nvmlInit()
      inspect(version=nvml.nvmlSystemGetDriverVersion())
      inspect(count=nvml.nvmlDeviceGetCount())
  
      for i in range(nvml.nvmlDeviceGetCount()):
          test_nvidia_device(i)
  
      nvml.nvmlShutdown()
  
imports: from labml import lab, logger
  from labml.logger import inspect
label_desc: Use image node_id 2 for calling a global function with example usage: test_nvidia() without return types