diff --git a/scripts/rknn-convert-tool/create_onnx.py b/scripts/rknn-convert-tool/create_onnx.py
index c568e2d57..c9b3b1133 100644
--- a/scripts/rknn-convert-tool/create_onnx.py
+++ b/scripts/rknn-convert-tool/create_onnx.py
@@ -82,7 +82,8 @@ def run_onnx_conversion_yolov5(model_path):
             "-r",
             os.path.join(ultralytics_folder_name_yolov5, "requirements.txt"),
             "torch<2.6.0",
-            "onnx",
+            "onnx==1.18.0",
+            "onnxscript",
         ]
     )
 
@@ -121,7 +122,9 @@ def run_onnx_conversion_yolov5(model_path):
 
 def run_onnx_conversion_no_anchor(model_path):
     check_or_clone_rockchip_repo(yolo_non_anchor_repo)
-    run_pip_install_or_else_exit(["-e", ultralytics_default_folder_name, "onnx"])
+    run_pip_install_or_else_exit(
+        ["-e", ultralytics_default_folder_name, "onnx==1.18.0", "onnxscript"]
+    )
 
     sys.path.insert(0, os.path.abspath(ultralytics_default_folder_name))
     model_abs_path = os.path.abspath(model_path)
diff --git a/scripts/rknn-convert-tool/rknn_conversion.ipynb b/scripts/rknn-convert-tool/rknn_conversion.ipynb
index e330677df..eb07261ae 100644
--- a/scripts/rknn-convert-tool/rknn_conversion.ipynb
+++ b/scripts/rknn-convert-tool/rknn_conversion.ipynb
@@ -109,7 +109,7 @@
     "\n",
     "#### Automatic installation\n",
     "\n",
-    "Please run `pip` below. If it does not work, refer to the instructions for manual installation.\n"
+    "Please run `pip` below. If it does not work, refer to the instructions for manual installation. You may need to restart your session after running the command below.\n"
    ]
   },
   {