def METHOD_NAME():
    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