AttributeError: module ‘onnxruntime‘ has no attribute ‘InferenceSession‘

万丈高楼平地起~

个人使用的机器上装有onnxruntime包，onnxruntime-gpu的包，预测速度很慢；

猜测应该是没用上onnxruntime-gpu，故卸载onnxruntime；

运行，又爆出bug，如标题；

无计可施，只能寄希望于重装大法：

pip uninstall onnxruntime-gpu
pip install onnxruntime-gpu

然后就fix 了这个bug；

cpu要装onnxruntime，GPU机器一定要装onnxruntime-gpu！

有问题随时交流~

彩蛋 

更多干货内容请关注公众号SLearningAI，欢迎CSDN与公众号同步交流^_^