wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 21:30:15 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix CUDA package with no drivers

Browse Source
This commit is contained in:
Benjamin Berman 2024-10-14 22:56:21 -07:00
parent f1ce08e68f
commit 9c9df424b4
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@ -1165,6 +1165,9 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma
if props.major < 6: if props.major < 6:
return False return False
except (ValueError, RuntimeError):
logging.warning("No CUDA devices were present, even though CUDA is available in this torch installation. This assumes the CPU device will be selected for computation")
return False
except AssertionError: except AssertionError:
logging.warning("Torch was not compiled with cuda support") logging.warning("Torch was not compiled with cuda support")
return False return False
@ -1226,7 +1229,7 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
props_major = min(torch.cuda.get_device_properties(torch.device(f"cuda:{i}")).major for i in range(torch.cuda.device_count())) props_major = min(torch.cuda.get_device_properties(torch.device(f"cuda:{i}")).major for i in range(torch.cuda.device_count()))
if props_major >= 8: if props_major >= 8:
return True return True
except ValueError: except (ValueError, RuntimeError):
logging.warning("No CUDA devices were present, even though CUDA is available in this torch installation. This assumes the CPU device will be selected for computation") logging.warning("No CUDA devices were present, even though CUDA is available in this torch installation. This assumes the CPU device will be selected for computation")
return False return False
except AssertionError: except AssertionError: