diff options
| author | pepe10-gpu <pepe.dannyboy@gmail.com> | 2022-11-07 01:05:51 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-07 01:05:51 +0000 |
| commit | cd6c55c1ab14fcab15329cde599cf79e8d555657 (patch) | |
| tree | 440e455fe57d173946e0f5231f717cbf2c00cd40 /modules/devices.py | |
| parent | 804d9fb83d0c63ca3acd36378707ce47b8f12599 (diff) | |
| download | stable-diffusion-webui-gfx803-cd6c55c1ab14fcab15329cde599cf79e8d555657.tar.gz stable-diffusion-webui-gfx803-cd6c55c1ab14fcab15329cde599cf79e8d555657.tar.bz2 stable-diffusion-webui-gfx803-cd6c55c1ab14fcab15329cde599cf79e8d555657.zip | |
16xx card fix
cudnn
Diffstat (limited to 'modules/devices.py')
| -rw-r--r-- | modules/devices.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/devices.py b/modules/devices.py index 7511e1dc..858bf399 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -39,10 +39,13 @@ def torch_gc(): def enable_tf32(): if torch.cuda.is_available(): + torch.backends.cudnn.benchmark = True + torch.backends.cudnn.enabled = True torch.backends.cuda.matmul.allow_tf32 = True torch.backends.cudnn.allow_tf32 = True + errors.run(enable_tf32, "Enabling TF32") device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None |