Skip to content

Commit cd6c55c

Browse files
authored
16xx card fix
cudnn
1 parent 804d9fb commit cd6c55c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

modules/devices.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,13 @@ def torch_gc():
3939

4040
def enable_tf32():
4141
if torch.cuda.is_available():
42+
torch.backends.cudnn.benchmark = True
43+
torch.backends.cudnn.enabled = True
4244
torch.backends.cuda.matmul.allow_tf32 = True
4345
torch.backends.cudnn.allow_tf32 = True
4446

4547

48+
4649
errors.run(enable_tf32, "Enabling TF32")
4750

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

0 commit comments

Comments
 (0)