my environment is as below
torch 2.2.2
onnx 1.18.0
onnxruntime 1.20.0
when I exporot onnx model
torch.onnx.export(
post_model,
rpn_input,
args.save_onnx,
export_params=True,
opset_version=21,
do_constant_folding=True,
keep_initializers_as_inputs=False,
input_names=['input'],
output_names=output_names,
)
it prompts error
2025-08-01 14:47:28,899 INFO ==> Done (loaded 201/201)
[INFO] Exporting 2 detection heads with heads: ['center', 'height', 'dim', 'rot']
[INFO] Exporting ONNX model to models/pcdet_neck_head.onnx, dtype=float32
/home/hitbuyi/anaconda3/envs/pt222/lib/python3.10/site-packages/torch/onnx/utils.py:1548: OnnxExporterWarning: Exporting to ONNX opset version 20 is not supported. by 'torch.onnx.export()'. The highest opset version supported is 17. To use a newer opset version, consider 'torch.onnx.dynamo_export()'. Note that dynamo_export() is in preview. Please report errors with dynamo_export() as Github issues to https://github.com/pytorch/pytorch/issues.
warnings.warn(
Traceback (most recent call last):
File "/home/hitbuyi/AD_Projects/Lidar/centerpoint_deploy/centerpoit_export/export_neck_head.py", line 195, in <module>
main()
File "/home/hitbuyi/AD_Projects/Lidar/centerpoint_deploy/centerpoit_export/export_neck_head.py", line 173, in main
torch.onnx.export(
File "/home/hitbuyi/anaconda3/envs/pt222/lib/python3.10/site-packages/torch/onnx/utils.py", line 516, in export
_export(
File "/home/hitbuyi/anaconda3/envs/pt222/lib/python3.10/site-packages/torch/onnx/utils.py", line 1576, in _export
GLOBALS.export_onnx_opset_version = opset_version
File "/home/hitbuyi/anaconda3/envs/pt222/lib/python3.10/site-packages/torch/onnx/_globals.py", line 59, in export_onnx_opset_version
raise ValueError(f"Unsupported ONNX opset version: {value}")
ValueError: Unsupported ONNX opset version: 21
Does pytorch 2.2.2 + onnx 1.18.0+ onnruntime 1.20.0 support opset 21?