Hotfix Geforce 16 series
This commit is contained in:
@@ -15,7 +15,8 @@ def test_create_inference_execution_providers() -> None:
|
|||||||
[
|
[
|
||||||
('CUDAExecutionProvider',
|
('CUDAExecutionProvider',
|
||||||
{
|
{
|
||||||
'device_id': '1'
|
'device_id': '1',
|
||||||
|
'cudnn_conv_algo_search': 'EXHAUSTIVE'
|
||||||
}),
|
}),
|
||||||
'CPUExecutionProvider'
|
'CPUExecutionProvider'
|
||||||
]
|
]
|
||||||
|
|||||||
Reference in New Issue
Block a user