Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

update cudnn

Browse Source
This commit is contained in:
Vladimir Mandic
2023-05-13 11:52:31 -04:00
parent d96ab6a1ae
commit a2923064a5
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+2 -6
View File
@@ -103,12 +103,8 @@ def set_cuda_params():
pass
if torch.backends.cudnn.is_available():
try:
if any([torch.cuda.get_device_capability(devid) == (7, 5) for devid in range(0, torch.cuda.device_count())]): # monkey-patch for old nvidia cards
torch.backends.cudnn.benchmark = True
torch.backends.cudnn.benchmark_limit = 1
else:
torch.backends.cudnn.benchmark = shared.opts.cudnn_benchmark
torch.backends.cudnn.benchmark_limit = 0
torch.backends.cudnn.benchmark = True
torch.backends.cudnn.benchmark_limit = 0 if shared.opts.cudnn_benchmark else 1
torch.backends.cudnn.allow_tf32 = shared.opts.cuda_allow_tf32
except:
pass