We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 804d9fb commit cd6c55cCopy full SHA for cd6c55c
modules/devices.py
@@ -39,10 +39,13 @@ def torch_gc():
39
40
def enable_tf32():
41
if torch.cuda.is_available():
42
+ torch.backends.cudnn.benchmark = True
43
+ torch.backends.cudnn.enabled = True
44
torch.backends.cuda.matmul.allow_tf32 = True
45
torch.backends.cudnn.allow_tf32 = True
46
47
48
+
49
errors.run(enable_tf32, "Enabling TF32")
50
51
device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None
0 commit comments