can print onnx providers

Author: xxaier

onnx/test/onnx/onnx_load.py

@@ -3,6 +3,7 @@
 import onnxruntime
 from misc.config import ONNX_FP
 from os.path import join
+from onnxruntime import get_all_providers
 
 session = onnxruntime.SessionOptions()
 option = onnxruntime.RunOptions()
@@ -14,10 +15,11 @@ def onnx_load(kind):
 
   sess = onnxruntime.InferenceSession(fp,
                                       sess_options=session,
-                                      providers=[
-                                          'TensorrtExecutionProvider',
-                                          'CUDAExecutionProvider',
-                                          'CoreMLExecutionProvider',
-                                          'CPUExecutionProvider'
-                                      ])
+                                      providers=get_all_providers())
   return sess
+
+
+if __name__ == '__main__':
+  sess = onnx_load('Txt')
+  providers = sess.get_providers()
+  print('provider', providers)