From 0c890b50e0f9e27bb2307d4cffec46a3fc16c4f2 Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Thu, 20 Mar 2025 23:03:23 +0900 Subject: [PATCH] proper zluda detection --- modules/devices.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/devices.py b/modules/devices.py index bd84ff7d4..b2a488d7b 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -52,7 +52,8 @@ def has_zluda() -> bool: return False try: dev = torch.device("cuda") - return torch.cuda.get_device_name(dev).endswith("[ZLUDA]") + cc = torch.cuda.get_device_capability(dev) + return cc == (8, 8) except Exception: return False