search

microsoft / nnfusion

A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
MIT License
958 stars 162 forks source link

[BUG] Error if running nnfusion with onnxruntime >= 1.9 #388

Closed mzmssg closed 2 years ago

mzmssg commented 2 years ago

🐛 Bug

ort >= 1.9 required to explicitly set provider, we need to upgrade our script or fix dependency version.

ValueError: This ORT build has ['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider'] enabled. Since ORT 1.9, you are required to explicitly set the providers parameter when instantiating InferenceSession. For example, onnxruntime.InferenceSession(..., providers=['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider'], ...)