[Python Compiling] Add option for python version (#3654)

lite/tools/build.sh

@@ -395,6 +395,7 @@ 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
@@ -530,6 +531,10 @@ function main {
                 PYTHON_EXECUTABLE_OPTION="-DPYTHON_EXECUTABLE=${i#*=}"
                 shift
                 ;;
+            --python_version=*)
+                PY_VERSION="${i#*=}"
+                shift
+                ;;
             --build_apu=*)
                 BUILD_APU="${i#*=}"
                 shift