16xx card fix

cudnn
2022-11-06 17:05:51 -08:00 · 2022-11-06 17:05:51 -08:00 · cd6c55c1ab
parent 804d9fb83d
commit cd6c55c1ab
1 changed files with 3 additions and 0 deletions
--- a/modules/devices.py
+++ b/modules/devices.py
@ -39,10 +39,13 @@ def torch_gc():

 def enable_tf32():
    if torch.cuda.is_available():
+        torch.backends.cudnn.benchmark = True
+        torch.backends.cudnn.enabled = True
        torch.backends.cuda.matmul.allow_tf32 = True
        torch.backends.cudnn.allow_tf32 = True


+
 errors.run(enable_tf32, "Enabling TF32")

 device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None