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

16xx card fix

cudnn
parent 804d9fb8
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment