pip install onnxruntime-gpu 的版本兼容问题

Win10 下Pytorch  1.9.0+cu102  安装  onnxruntime-gpu 后运行到onnx模型加载推理后一直提示加载库错误

[E:onnxruntime:Default, provider_bridge_ort.cc:952 onnxruntime::ProviderLibrary::Get] Failed to load library, error code: 126

查看onnxruntime-gpu的版本是1.8.1，经过分析这类错误大多和版本相关。

pip install onnxruntime-gpu == 1.5.2

 

 

经过一阵尝试，降低onnxruntime-gpu的版本，不再报程序错误了。