fix setup

ecmwf-lab · Sep 12, 2023 · 6ac12b3 · 6ac12b3
1 parent f0e788f
commit 6ac12b3
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/setup.py b/setup.py
@@ -47,12 +47,7 @@ def check_gpus():
 
 
 def has_gpu():
-    try:
-        import GPUtil
-
-        return len(GPUtil.getAvailable()) > 0
-    except ImportError:
-        return check_gpus() > 0
+    return check_gpus() > 0
 
 
 onnxruntime = "onnxruntime"
@@ -82,7 +77,6 @@ def has_gpu():
         "ai-models",
         "onnx",
         os.environ.get("ONNXRUNTIME", onnxruntime),
-        "GPUtil",
     ],
     zip_safe=True,
     keywords="tool",