Unverified Commit cd6c55c1 authored by pepe10-gpu's avatar pepe10-gpu Committed by GitHub
Browse files

16xx card fix

cudnn
parent 804d9fb8
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -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