diff --git a/python/gen_requirements.py b/python/gen_requirements.py
index e9f3772ee733..47be2613a6ab 100755
--- a/python/gen_requirements.py
+++ b/python/gen_requirements.py
@@ -109,6 +109,7 @@
             [
                 "future",  # Hidden dependency of torch.
                 "onnx",
+                "onnxoptimizer",
                 "onnxruntime",
                 "torch",
                 "torchvision",
@@ -145,6 +146,7 @@
             [
                 "future",  # Hidden dependency of torch.
                 "onnx",
+                "onnxoptimizer",
                 "onnxruntime",
                 "paddlepaddle",
                 "tensorflow",
@@ -239,6 +241,7 @@
     ("matplotlib", None),
     ("numpy", None),
     ("onnx", None),
+    ("onnxoptimizer", None),
     ("onnxruntime", None),
     ("opencv-python", None),
     ("paddlepaddle", None),