Autoselect CUDA Device (#56)

* sort by free ram * debug value changed * fix * delete debug device
supervisely-ecosystem · Nov 7, 2024 · 375baf2 · 375baf2
1 parent 2fa5cf2
commit 375baf2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train/src/ui/monitoring.py b/train/src/ui/monitoring.py
@@ -80,7 +80,7 @@ def init_devices():
             "free": free_mem,
         }
         devices.append(device_info)
-    return devices
+    return sorted(devices, key=lambda x: x["free"], reverse=True)
 
 
 def init_chart(