Commit 2651267e authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix #4407 breaking UI entirely for card other than ones related to the PR

parent ce049c47
Loading
Loading
Loading
Loading
+2 −4
Original line number Diff line number Diff line
@@ -53,12 +53,10 @@ def torch_gc():

def enable_tf32():
    if torch.cuda.is_available():
        for devid in range(0,torch.cuda.device_count()):
            if torch.cuda.get_device_capability(devid) == (7, 5):
                shd = True
        if shd:
        if any([torch.cuda.get_device_capability(devid) == (7, 5) for devid in range(0, torch.cuda.device_count())]):
            torch.backends.cudnn.benchmark = True
            torch.backends.cudnn.enabled = True

        torch.backends.cuda.matmul.allow_tf32 = True
        torch.backends.cudnn.allow_tf32 = True