From 2c47382b3705faaf3a81e2d6b7ecc40481ad17aa Mon Sep 17 00:00:00 2001 From: Grzegorz Klimaszewski <166530809+grzegorz-roboflow@users.noreply.github.com> Date: Tue, 3 Dec 2024 16:46:42 +0100 Subject: [PATCH] formatting --- inference/core/devices/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference/core/devices/utils.py b/inference/core/devices/utils.py index f1ca572bd..b4d7891f2 100644 --- a/inference/core/devices/utils.py +++ b/inference/core/devices/utils.py @@ -25,7 +25,7 @@ def get_gpu_id(): Optional[int]: GPU ID if available, None otherwise. """ try: - from pynvml import nvmlInit, nvmlDeviceGetCount + from pynvml import nvmlDeviceGetCount, nvmlInit nvmlInit() gpus_count = nvmlDeviceGetCount()