Add warning for old pytorch. (#11718)

This commit is contained in:
comfyanonymous 2026-01-07 18:07:26 -08:00 committed by GitHub
parent b6c79a648a
commit 21e8425087
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -19,6 +19,7 @@ try:
cuda_version = tuple(map(int, str(torch.version.cuda).split('.')))
if cuda_version < (13,):
ck.registry.disable("cuda")
logging.warning("WARNING: You need pytorch with cu130 or higher to use optimized CUDA operations.")
ck.registry.disable("triton")
for k, v in ck.list_backends().items():