浏览代码

add comment for #4407 and remove seemingly unnecessary cudnn.enabled

AUTOMATIC 2 年之前
父节点
当前提交
46b0d230e7
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 3 1
      modules/devices.py

+ 3 - 1
modules/devices.py

@@ -53,9 +53,11 @@ def torch_gc():
 
 
 def enable_tf32():
 def enable_tf32():
     if torch.cuda.is_available():
     if torch.cuda.is_available():
+
+        # enabling benchmark option seems to enable a range of cards to do fp16 when they otherwise can't
+        # see https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/4407
         if any([torch.cuda.get_device_capability(devid) == (7, 5) for devid in range(0, torch.cuda.device_count())]):
         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.benchmark = True
-            torch.backends.cudnn.enabled = True
 
 
         torch.backends.cuda.matmul.allow_tf32 = True
         torch.backends.cuda.matmul.allow_tf32 = True
         torch.backends.cudnn.allow_tf32 = True
         torch.backends.cudnn.allow_tf32 = True