Add env TORCH_AMD_CUDNN_ENABLED (import os)

This commit is contained in:
Christopher Anderson 2025-11-08 16:12:45 +11:00
parent 417dbcb32f
commit 90ba86a95f

View File

@ -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