diff --git a/tools/Dockerfile.runtime_template b/tools/Dockerfile.runtime_template
index 9b78b101eadbec6ce48212155d2a43efa31bb488..1aa703ab6bb8efff89faade35821a7ce4d7f2a48 100644
--- a/tools/Dockerfile.runtime_template
+++ b/tools/Dockerfile.runtime_template
@@ -28,7 +28,7 @@ WORKDIR /home
 # install whl and bin
 WORKDIR /home
       COPY tools/dockerfiles/build_scripts /build_scripts
-      RUN bash /build_scripts/install_whl.sh 0.5.0 2.0.0 <<run_env>> <<python_version>> && rm -rf /build_scripts
+      RUN bash /build_scripts/install_whl.sh <<serving_version>> <<paddle_version>> <<run_env>> <<python_version>> && rm -rf /build_scripts
 
 # install tensorrt
 WORKDIR /home