aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/ci_build/protobuf/protobuf_optimized_pip.sh
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/tools/ci_build/protobuf/protobuf_optimized_pip.sh')
-rwxr-xr-xtensorflow/tools/ci_build/protobuf/protobuf_optimized_pip.sh3
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/tools/ci_build/protobuf/protobuf_optimized_pip.sh b/tensorflow/tools/ci_build/protobuf/protobuf_optimized_pip.sh
index 48f3dcdf7a..078aed7154 100755
--- a/tensorflow/tools/ci_build/protobuf/protobuf_optimized_pip.sh
+++ b/tensorflow/tools/ci_build/protobuf/protobuf_optimized_pip.sh
@@ -15,6 +15,7 @@
# ==============================================================================
PROTOBUF_VERSION="3.1.0"
+PYTHON_BIN=${PYTHON_BIN:-python}
DIR=${PWD}/protobuf
set -ex
@@ -29,4 +30,4 @@ CXXFLAGS="-fPIC -g -O2" ./configure
make -j8
export PROTOC=$DIR/src/protoc
cd python
-python setup.py bdist_wheel --cpp_implementation --compile_static_extension
+$PYTHON_BIN setup.py bdist_wheel --cpp_implementation --compile_static_extension