Log the pytorch version.

comfyanonymous · comfyanonymous · commit 09e069ae6ca7 · 2024-05-20T06:22:29.000-04:00
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -120,6 +120,11 @@ def get_total_memory(dev=None, torch_total_too=False):
 total_ram = psutil.virtual_memory().total / (1024 * 1024)
 logging.info("Total VRAM {:0.0f} MB, total RAM {:0.0f} MB".format(total_vram, total_ram))
 
+try:
+    logging.info("pytorch version: {}".format(torch.version.__version__))
+except:
+    pass
+
 try:
     OOM_EXCEPTION = torch.cuda.OutOfMemoryError
 except:
