Disable comfy kitchen cuda if pytorch cuda less than 13 (#11681)

This commit is contained in:
comfyanonymous
2026-01-06 19:13:43 -08:00
committed by GitHub
parent 2c03884f5f
commit edee33f55e

View File

@@ -13,6 +13,13 @@ try:
get_layout_class,
)
_CK_AVAILABLE = True
if torch.version.cuda is None:
ck.registry.disable("cuda")
else:
cuda_version = tuple(map(int, str(torch.version.cuda).split('.')))
if cuda_version < (13,):
ck.registry.disable("cuda")
ck.registry.disable("triton")
for k, v in ck.list_backends().items():
logging.info(f"Found comfy_kitchen backend {k}: {v}")