diff --git a/frigate/util/model.py b/frigate/util/model.py index 5e18ed828..685cd34ec 100644 --- a/frigate/util/model.py +++ b/frigate/util/model.py @@ -28,10 +28,14 @@ def get_ort_providers( providers = [] options = [] - for provider in providers: - if provider == "CudaExecutionProvider": + for provider in ort.get_available_providers(): + if provider == "CUDAExecutionProvider": providers.append(provider) - options.append({}) + options.append( + { + "arena_extend_strategy": "kSameAsRequested", + } + ) elif provider == "TensorrtExecutionProvider": # TensorrtExecutionProvider uses too much memory without options to control it pass