changed gpus list coming from habitat (#33)

CentML · Feb 8, 2023 · aa12301 · aa12301
1 parent 022462c
commit aa12301
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/skyline/analysis/session.py b/skyline/analysis/session.py
@@ -200,12 +200,17 @@ def habitat_predict(self):
 
         print("habitat_predict: begin")
         DEVICES = [
-            habitat.Device.P4000,
             habitat.Device.P100,
-            habitat.Device.V100,
-            habitat.Device.T4,
+            habitat.Device.P4000,
             habitat.Device.RTX2070,
             habitat.Device.RTX2080Ti,
+            habitat.Device.T4,
+            habitat.Device.V100,
+            habitat.Device.A100,
+            habitat.Device.RTX3090,
+            habitat.Device.A40,
+            habitat.Device.A4000,
+            habitat.Device.RTX4000
         ]
 
         # Detect source GPU