增加windows 实现 (#671)

增加Windows CPU信息获取
eosphoros-ai · Oct 13, 2023 · 867c2f2 · 867c2f2
2 parents f1bb53f + 300b01a
commit 867c2f2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/pilot/utils/system_utils.py b/pilot/utils/system_utils.py
@@ -205,8 +205,14 @@ def get_cpu_info():
         except:
             pass
     elif os_type == OSType.WINDOWS:
-        # TODO
-        raise NotImplementedError
+        try:
+            output = subprocess.check_output("wmic cpu get Name", shell=True).decode(
+                "utf-8"
+            )
+            lines = output.splitlines()
+            cpu_info = lines[2].split(":")[-1].strip()
+        except:
+            pass
 
     return os_type, avx_type, cpu_info, distribution