Fix gpu memory info

LibrePhotos · Sep 12, 2023 · 9d2525a · 9d2525a
1 parent fc9be2b
commit 9d2525a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/api_util.py b/api/api_util.py
@@ -316,7 +316,7 @@ def get_server_stats():
 
     if torch.cuda.is_available():
         gpu_name = torch.cuda.get_device_name(0)
-        gpu_memory = torch.cuda.memory_summary()
+        gpu_memory = calc_megabytes(torch.cuda.get_device_properties(0).total_memory)
     else:
         gpu_name = ""
         gpu_memory = ""