code: def test_get_nvidia_gpu_stats():
      current_device = torch.device(
          f"cuda:{torch.cuda.current_device()}"
      )
      gpu_stats = get_nvidia_gpu_stats(current_device)
      fields = [
          "utilization.gpu",
          "memory.used",
          "memory.free",
          "utilization.memory",
      ]
  
      for f in fields:
          assert any(f in h for h in gpu_stats)
  
file_path: pytorch-lightning/tests/tests_pytorch/accelerators/test_gpu.py
imports: from unittest import mock
  import pytest
  import torch
  from lightning.pytorch import Trainer
  from lightning.pytorch.accelerators import CUDAAccelerator
  from lightning.pytorch.accelerators.cuda import get_nvidia_gpu_stats
  from lightning.pytorch.demos.boring_classes import BoringModel
  from tests_pytorch.helpers.runif import RunIf
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_nvidia_gpu_stats() without return types