From 417dbcb32f3ffa49fd9d1952c65fedfd94aaa356 Mon Sep 17 00:00:00 2001 From: Christopher Anderson Date: Sat, 8 Nov 2025 03:37:46 +1100 Subject: [PATCH 1/3] Add env TORCH_AMD_CUDNN_ENABLED --- comfy/model_management.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 7012df858..949d21dbd 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -338,8 +338,11 @@ try: if is_amd(): arch = torch.cuda.get_device_properties(get_torch_device()).gcnArchName if not (any((a in arch) for a in AMD_RDNA2_AND_OLDER_ARCH)): - torch.backends.cudnn.enabled = False # Seems to improve things a lot on AMD - logging.info("Set: torch.backends.cudnn.enabled = False for better AMD performance.") + torch.backends.cudnn.enabled = os.environ.get("TORCH_AMD_CUDNN_ENABLED", "0").strip().lower() not in { + "0", "off", "false", "disable", "disabled", "no"} + if not torch.backends.cudnn.enabled: + logging.info( + "ComfyUI has set torch.backends.cudnn.enabled to False for better AMD performance. Set environment var TORCH_AMD_CUDDNN_ENABLED=1 to enable it again.") try: rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2])) From 90ba86a95fd78e86cdf3ada7df2983340d92096e Mon Sep 17 00:00:00 2001 From: Christopher Anderson Date: Sat, 8 Nov 2025 16:12:45 +1100 Subject: [PATCH 2/3] Add env TORCH_AMD_CUDNN_ENABLED (import os) --- comfy/model_management.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/model_management.py b/comfy/model_management.py index 949d21dbd..8c2aa5179 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -26,6 +26,7 @@ import importlib import platform import weakref import gc +import os class VRAMState(Enum): DISABLED = 0 #No vram present: no need to move models to vram From de408d5f2c07a2fd27d7c2e245eed1e244f95a43 Mon Sep 17 00:00:00 2001 From: Christopher Anderson Date: Sat, 8 Nov 2025 20:45:32 +1100 Subject: [PATCH 3/3] Add env TORCH_AMD_CUDNN_ENABLED (fix typo) --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 8c2aa5179..2c5f672f9 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -343,7 +343,7 @@ try: "0", "off", "false", "disable", "disabled", "no"} if not torch.backends.cudnn.enabled: logging.info( - "ComfyUI has set torch.backends.cudnn.enabled to False for better AMD performance. Set environment var TORCH_AMD_CUDDNN_ENABLED=1 to enable it again.") + "ComfyUI has set torch.backends.cudnn.enabled to False for better AMD performance. Set environment var TORCH_AMD_CUDNN_ENABLED=1 to enable it again.") try: rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2]))