From 4183e71cf83f744424bab978c12c4effd461c086 Mon Sep 17 00:00:00 2001 From: Pikachu~~~ Date: Wed, 31 May 2023 10:06:42 +0800 Subject: [PATCH] fix TensorRTExcutionProvider errors --- globals.py | 7 +++++++ 1 file changed, 7 insertions(+) create mode 100644 globals.py diff --git a/globals.py b/globals.py new file mode 100644 index 0000000..3525c9d --- /dev/null +++ b/globals.py @@ -0,0 +1,7 @@ +import onnxruntime + +use_gpu = False +providers = onnxruntime.get_available_providers() + +if 'TensorrtExecutionProvider' in providers: + providers.remove('TensorrtExecutionProvider')