From f000dfddbe43ef753dc8355fc16e98c9bb680f87 Mon Sep 17 00:00:00 2001
From: Changming Sun <chasun@microsoft.com>
Date: Wed, 23 Jun 2021 15:50:03 -0700
Subject: [PATCH] Update run_dockerbuild.sh: set default python version based
 on OS version (#8136)

---
 tools/ci_build/github/linux/run_dockerbuild.sh | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/tools/ci_build/github/linux/run_dockerbuild.sh b/tools/ci_build/github/linux/run_dockerbuild.sh
index 117b77ebbc..a2b34144da 100755
--- a/tools/ci_build/github/linux/run_dockerbuild.sh
+++ b/tools/ci_build/github/linux/run_dockerbuild.sh
@@ -48,7 +48,15 @@ esac
 done
 
 EXIT_CODE=1
-PYTHON_VER=${PYTHON_VER:=3.6}
+if [ $BUILD_OS = "ubuntu18.04" ]; then
+   DEFAULT_PYTHON_VER="3.6"
+elif [ $BUILD_OS = "ubuntu20.04" ]; then
+   DEFAULT_PYTHON_VER="3.8"
+else
+   DEFAULT_PYTHON_VER="3.6"   
+fi
+		
+PYTHON_VER=${PYTHON_VER:=$DEFAULT_PYTHON_VER}
 echo "bo=$BUILD_OS bd=$BUILD_DEVICE bdir=$BUILD_DIR pv=$PYTHON_VER bex=$BUILD_EXTR_PAR"
 
 GET_DOCKER_IMAGE_CMD="${SOURCE_ROOT}/tools/ci_build/get_docker_image.py"