jiyuhang 3 месяцев назад
Родитель
Сommit
031da73044
2 измененных файлов с 7 добавлено и 7 удалено
  1. 1 1
      app.py
  2. 6 6
      app/algorithm/monitor/monitor_util.py

+ 1 - 1
app.py

@@ -6,4 +6,4 @@ app = create_app()
 
 
 if __name__ == '__main__':
-    app.run(port=5000, debug=True)
+    app.run(port=5000, debug=False)

+ 6 - 6
app/algorithm/monitor/monitor_util.py

@@ -152,12 +152,12 @@ class Monitor:
                         except:
                             power_watts = 0
                         device_id = f'gpu_{i}_'
-                        self.usage[device_id+'mem'] = memory_mb
-                        self.usage[device_id+'rate_gpu'] = gpu_util.gpu if gpu_util else 0
-                        # self.usage[device_id+'rate_mem'] = gpu_util.memory if gpu_util else 0
-                        self.usage[device_id+'temperature'] = temperature
-                        self.usage[device_id+'power'] = power_watts
-                        print(f"GPU {i}:")
+                        self.usage[device_id+'mem'] = memory_mb  # 显存 MB
+                        self.usage[device_id+'rate_gpu'] = gpu_util.gpu if gpu_util else 0  # GPU利用率 %
+                        self.usage[device_id+'rate_mem'] = gpu_util.memory if gpu_util else 0  # 显存利用率 %
+                        self.usage[device_id+'temperature'] = temperature  # 显卡温度 °C
+                        self.usage[device_id+'power'] = power_watts  # 显卡功耗 W
+                        # print(f"GPU {i}:")
                         # print(f"  内存使用: {memory_mb:.2f} MB")
                         # print(f"  GPU 利用率: {gpu_util.gpu if gpu_util else 0}%")
                         # print(f"  内存利用率: {gpu_util.memory if gpu_util else 0}%")