code: def test_1_visible_devices():
      if nvml.device_get_count() < 1:
          pytest.skip("No GPUs available")
  
      os.environ["CUDA_VISIBLE_DEVICES"] = "0"
      output = nvml.one_time()
      h = nvml._pynvml_handles()
      assert (
          output["memory-total"]
          == pynvml.nvmlDeviceGetMemoryInfo(h).total
      )
  
file_path: distributed/distributed/diagnostics/tests/test_nvml.py
imports: from __future__ import annotations
  import multiprocessing
  import os
  import pytest
  import dask
  from distributed.diagnostics import nvml
  from distributed.utils_test import gen_cluster
label_desc: Use image node_id 7 for calling a global function with example usage: test_1_visible_devices() without return types