additional python 3.12 compatibility

2026-09-20 01:31:13 +02:00 · 2024-06-08 14:14:48 -04:00
parent d4fead09a6
commit 6d6f1de295
7 changed files with 9 additions and 6 deletions
@@ -235,7 +235,7 @@ def set_cuda_params():
                torch.use_deterministic_algorithms(shared.opts.cudnn_deterministic)
                log.debug(f'Torch mode: deterministic={shared.opts.cudnn_deterministic}')
                if shared.opts.cudnn_deterministic:
-                    os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
+                    os.environ.setdefault('CUBLAS_WORKSPACE_CONFIG', ':4096:8')
                torch.backends.cudnn.benchmark = True
                if shared.opts.cudnn_benchmark:
                    log.debug('Torch cuDNN: enable benchmark')