Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "13994b2d3fbb536044ae1d4befb8102bfa6424ec"
Commit 09e069ae authored by comfyanonymous's avatar comfyanonymous
Browse files

Log the pytorch version.

parent 11a2ad51
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
comfy/model_management.py
View file @ 09e069ae
...@@ -120,6 +120,11 @@ total_vram = get_total_memory(get_torch_device()) / (1024 * 1024) ...@@ -120,6 +120,11 @@ total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
total_ram = psutil.virtual_memory().total / (1024 * 1024) 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)) 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: try:
OOM_EXCEPTION = torch.cuda.OutOfMemoryError OOM_EXCEPTION = torch.cuda.OutOfMemoryError
except: except:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment