dep_check.py now ignores GPUtil

2024-03-11 18:55:37 +08:00
parent 91eb72e515
commit dd89041f72
2 changed files with 5 additions and 3 deletions
--- a/backend-python/routes/config.py
+++ b/backend-python/routes/config.py
@@ -127,9 +127,12 @@ def update_config(body: ModelConfigBody):

@router.get("/status", tags=["Configs"])
 def status():
-    import GPUtil
+    try:
+        import GPUtil

-    gpus = GPUtil.getGPUs()
+        gpus = GPUtil.getGPUs()
+    except:
+        gpus = []
    if len(gpus) == 0:
        device_name = "CPU"
    else: