16xx card fix

cudnn
This commit is contained in:
pepe10-gpu 2022-11-06 17:05:51 -08:00 committed by GitHub
parent 804d9fb83d
commit cd6c55c1ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

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