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

fix cn

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-11-01 12:21:19 -04:00
parent 0e567d3dc5
commit 495cfd8632
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+3 -3
View File
@@ -429,14 +429,14 @@ def set_cudnn_params():
torch.use_deterministic_algorithms(opts.cudnn_deterministic)
if opts.cudnn_deterministic:
os.environ.setdefault('CUBLAS_WORKSPACE_CONFIG', ':4096:8')
log.debug('Torch cuDNN: deterministic=True')
log.debug(f'Torch cuDNN: deterministic={opts.cudnn_deterministic}')
torch.backends.cudnn.benchmark = opts.cudnn_benchmark
if opts.cudnn_benchmark:
log.debug('Torch cuDNN: benchmark=True')
log.debug(f'Torch cuDNN: benchmark={opts.cudnn_benchmark}')
torch.backends.cudnn.benchmark_limit = opts.cudnn_benchmark_limit
torch.backends.cudnn.allow_tf32 = True
except Exception as e:
log.warning(f'Torch cudnn: {e}')
log.warning(f'Torch cuDNN: {e}')
def override_ipex_math():