Hotfix Geforce 16 series

2025-01-02 22:13:31 +01:00
parent 197773c346
commit 1e3bab9644
1 changed files with 2 additions and 1 deletions
--- a/tests/test_execution.py
+++ b/tests/test_execution.py
@@ -15,7 +15,8 @@ def test_create_inference_execution_providers() -> None:
 	[
 		('CUDAExecutionProvider',
 		{
-			'device_id': '1'
+			'device_id': '1',
+			'cudnn_conv_algo_search': 'EXHAUSTIVE'
 		}),
 		'CPUExecutionProvider'
 	]