diff --git a/lite/tools/build.sh b/lite/tools/build.sh
index 02c1bb5e3642ee107708b63f19ffb12fede150c7..756d392a91699e2573415ab56fabbb1363d0d5ff 100755
--- a/lite/tools/build.sh
+++ b/lite/tools/build.sh
@@ -383,8 +383,8 @@ function make_x86 {
             -DLITE_WITH_XTCL=$BUILD_XTCL \
             -DXPU_SDK_ROOT=$XPU_SDK_ROOT \
             -DCMAKE_BUILD_TYPE=Release \
+            -DPY_VERSION=$PY_VERSION \
             $PYTHON_EXECUTABLE_OPTION
-
   make publish_inference -j$NUM_PROC
   cd -
 }
@@ -518,6 +518,10 @@ function main {
                 PYTHON_EXECUTABLE_OPTION="-DPYTHON_EXECUTABLE=${i#*=}"
                 shift
                 ;;
+            --python_version=*)
+                PY_VERSION="${i#*=}"
+                shift
+                ;;
             --build_apu=*)
                 BUILD_APU="${i#*=}"
                 shift