报错如下:
先说解决办法:将1.16.2版本的onnx降级为1.16.1
1)具体操作:
本人使用的是anaconda虚拟环境,故操作分两步:卸载旧版本onnx,安装指定版本
激活的虚拟环境中使用CLI命令 ,卸载原有版本1.16.2
pip uninstall onnx
然后使用命令安装onnx的1.16.1版本(其它低版本可能也行,本人没过多尝试)
pip install onnx==1.16.1
ok!此时运行命令,即可成功导出
yolo export model=yolov8n.pt format=onnx simplify=True
导出结果图:
2)conda虚拟环境中主要的库版本
python 3.8.0
torch 2.4.0+cu118
torchaudio 2.4.0+cu118
torchvision 0.19.0+cu118
ultralytics 8.2.74
onnxruntime 1.18.1
onnx-simplifier 0.4.36
onnxruntime-gpu 1.18.1
onnxsim 0.4.36
onnxslim 0.1.32
3)排错(踩坑)过程
重装ultralytics,重装onnx,onnxruntime,以及修改python版本等均没解决问题,之前以为是缺乏库或者依赖的问题,同样的环境更换为1.16.1版本onnx成功导出