Fix OpenVINO by aliasing GPU.0 to GPU
This commit is contained in:
parent
853474bf79
commit
20d2b6a4ea
@ -52,7 +52,7 @@ def create_execution_providers(execution_device_id : str, execution_provider_key
|
||||
if execution_provider_key == 'openvino':
|
||||
execution_providers.append((execution_provider_set.get(execution_provider_key),
|
||||
{
|
||||
'device_type': 'GPU' if execution_device_id == 0 else 'GPU.' + execution_device_id,
|
||||
'device_type': 'GPU' if execution_device_id == '0' else 'GPU.' + execution_device_id,
|
||||
'precision': 'FP32'
|
||||
}))
|
||||
if execution_provider_key in [ 'directml', 'rocm' ]:
|
||||
|
Loading…
Reference in New Issue
Block a user