mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove cudnn_conv_algo_search tweaks

Browse Source
This commit is contained in:
henryruhs 2024-11-04 09:36:39 +01:00
parent b05e25cf36
commit 696b54099e
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
facefusion/execution.py
View File

@ -36,8 +36,7 @@ def create_execution_providers(execution_device_id : str, execution_provider_key
if execution_provider_key == 'cuda':
execution_providers.append((execution_provider_set.get(execution_provider_key),
{
'device_id': execution_device_id,
'cudnn_conv_algo_search': 'EXHAUSTIVE' if use_exhaustive() else 'HEURISTIC'
'device_id': execution_device_id
}))
if execution_provider_key == 'tensorrt':
execution_providers.append((execution_provider_set.get(execution_provider_key),
@ -69,13 +68,6 @@ def create_execution_providers(execution_device_id : str, execution_provider_key
return execution_providers
def use_exhaustive() -> bool:
execution_devices = detect_static_execution_devices()
product_names = ('GeForce GTX 1630', 'GeForce GTX 1650', 'GeForce GTX 1660')
return any(execution_device.get('product').get('name').startswith(product_names) for execution_device in execution_devices)
def run_nvidia_smi() -> subprocess.Popen[bytes]:
commands = [ 'nvidia-smi', '--query', '--xml-format' ]
return subprocess.Popen(commands, stdout = subprocess.PIPE)