Merge pull request #5385 from gunan/master

Create a script for building tensorflow pip package on windows using bazel.
This commit is contained in:
gunan 2016-11-04 10:43:57 -07:00 committed by GitHub
commit 9ff79e057f
2 changed files with 57 additions and 0 deletions

View File

@ -0,0 +1,56 @@
#!/bin/bash
# This script assumes the standard setup on tensorflow Jenkins windows machines.
# It is NOT guaranteed to work on any other machine. Use at your own risk!
#
# REQUIREMENTS:
# * All installed in standard locations:
# - JDK8, and JAVA_HOME set.
# - Microsoft Visual Studio 2015 Community Edition
# - Msys2
# - Anaconda3
# * Bazel windows executable copied as "bazel.exe" and included in PATH.
# All commands shall pass, and all should be visible.
set -x
set -e
# This script is under <repo_root>/tensorflow/tools/ci_build/windows/cpu/pip/
# Change into repository root.
script_dir=$(dirname $0)
cd ${script_dir%%tensorflow/tools/ci_build/windows/cpu/pip}
# Use a temporary directory with a short name.
export TMPDIR="C:/tmp"
# Set bash path
export BAZEL_SH="C:/tools/msys64/usr/bin/bash"
# Set Python path for ./configure
export PYTHON_BIN_PATH="C:/Program Files/Anaconda3/python"
# Set Python path for cc_configure.bzl
export BAZEL_PYTHON="C:/Program Files/Anaconda3/python"
# Set Visual Studio path
export BAZEL_VS="C:/Program Files (x86)/Microsoft Visual Studio 14.0"
# Add python into PATH, it's needed because gen_git_source.py uses
# '/usr/bin/env python' as a shebang
export PATH="/c/Program Files/Anaconda3:$PATH"
# bazel clean --expunge doesn't work on Windows yet.
# Clean the output base manually to ensure build correctness
bazel clean
output_base=$(bazel info output_base)
bazel shutdown
rm -rf ${output_base}
echo "" | ./configure
bazel build -c opt --cpu=x64_windows_msvc --host_cpu=x64_windows_msvc\
--copt="/w" --verbose_failures --experimental_ui\
tensorflow/tools/pip_package:build_pip_package || exit $?
./bazel-bin/tensorflow/tools/pip_package/build_pip_package $PWD

View File

@ -0,0 +1 @@
c:\tools\msys64\usr\bin\bash -l %cd%/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh %*