module 'onnxruntime' has no attribute 'InferenceSession'

Rapternmn / PyTorch-Onnx-Tensorrt

A set of tool which would make your life easier with Tensorrt and Onnxruntime. This Repo is designed for YoloV3

80 stars 18 forks source link

Open Tian14267 opened 4 years ago

Tian14267 commented 4 years ago

I use pip install onnxruntime-gpu , and I get this error:

module 'onnxruntime' has no attribute 'InferenceSession'

when I use: sess = onnxruntime.InferenceSession(model_path)

Rapternmn commented 4 years ago

My onnxruntime version was '0.5.0'

What is your version?