diff --git a/paddleslim/common/load_model.py b/paddleslim/common/load_model.py
index 2f31f97d341007e6fa18069c0ff61a7a3bfd3ad5..0b4bef2c2ec369178f87d73be91cc6ff47dffdd2 100644
--- a/paddleslim/common/load_model.py
+++ b/paddleslim/common/load_model.py
@@ -217,11 +217,11 @@ def export_onnx(model_dir,
     try:
         import paddle2onnx
         version = paddle2onnx.__version__
-        if version != '1.0.0rc4':
-            os.system('python -m pip install -U paddle2onnx==1.0.0rc4')
+        if version != '1.0.1':
+            os.system('python -m pip install -U paddle2onnx==1.0.1')
     except:
         from pip._internal import main
-        main(['install', 'paddle2onnx==1.0.0rc4'])
+        main(['install', 'paddle2onnx==1.0.1'])
     import paddle2onnx
     paddle2onnx.command.c_paddle_to_onnx(
         model_file=os.path.join(model_dir, model_filename),