From 02c73f8faf80f1d6923cbf98422b2ce582cbbbee Mon Sep 17 00:00:00 2001
From: Xuehai Pan <XuehaiPan@pku.edu.cn>
Date: Fri, 22 Jul 2022 14:45:09 +0800
Subject: [PATCH] chore(select): sort device index in descending to keep <GPU
 0> free

Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
---
 nvitop/select.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nvitop/select.py b/nvitop/select.py
index 295a4bd..51fafd4 100644
--- a/nvitop/select.py
+++ b/nvitop/select.py
@@ -209,6 +209,7 @@ def select_devices(
             (not math.isnan(device.memory_used), -device.memory_used),  # descending
             (not math.isnan(device.gpu_utilization), device.gpu_utilization),  # ascending
             (not math.isnan(device.memory_utilization), device.memory_utilization),  # ascending
+            -device.physical_index,  # descending to keep <GPU 0> free
         ),
     )  # type: List[DeviceSnapshot]