Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Attempt 4 at fixing release builds #36085

Merged
merged 2 commits into from Jan 21, 2020
Merged

Attempt 4 at fixing release builds #36085

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
619c578
Disable the gpu on cpu tests as they were added for 2.1
mihaimaruseac Jan 21, 2020
63aedd7
Disable test that times out on mac non pip builds
mihaimaruseac Jan 21, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
1 change: 1 addition & 0 deletions tensorflow/python/kernel_tests/BUILD
View file
Open in desktop
Expand Up @@ -3275,6 +3275,7 @@ cuda_py_test(
"//tensorflow/python:framework_for_generated_wrappers",
],
tags = [
"no_mac", # timeouts on GPU
"no_pip",
"notap", # http://b/31080670
],
Expand Down
38 changes: 1 addition & 37 deletions tensorflow/tools/ci_build/release/ubuntu_16/gpu_pip_on_cpu/build.sh
View file
Open in desktop
Expand Up @@ -16,40 +16,4 @@
set -e
set -x

source tensorflow/tools/ci_build/release/common.sh
update_bazel_linux

install_ubuntu_16_pip_deps pip3.6

# Run configure.
export TF_NEED_GCP=1
export TF_NEED_HDFS=1
export TF_NEED_S3=1
export TF_NEED_CUDA=1
export TF_CUDA_VERSION=10
export TF_CUDNN_VERSION=7
export TF_NEED_TENSORRT=1
export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which python3.6)
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0

yes "" | "$PYTHON_BIN_PATH" configure.py

########################
## Build GPU pip package
########################
bazel build --config=opt \
--crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain \
tensorflow/tools/pip_package:build_pip_package

PIP_WHL_DIR=whl
mkdir -p ${PIP_WHL_DIR}
PIP_WHL_DIR=$(readlink -f ${PIP_WHL_DIR}) # Get absolute path
bazel-bin/tensorflow/tools/pip_package/build_pip_package "${PIP_WHL_DIR}"
WHL_PATH=$(ls "${PIP_WHL_DIR}"/*.whl)

cp "${WHL_PATH}" "$(pwd)"/.
chmod +x tensorflow/tools/ci_build/builds/docker_cpu_pip.sh
docker run -e "BAZEL_VERSION=${BAZEL_VERSION}" -e "CI_BUILD_USER=$(id -u -n)" -e "CI_BUILD_UID=$(id -u)" -e "CI_BUILD_GROUP=$(id -g -n)" -e "CI_BUILD_GID=$(id -g)" -e "CI_BUILD_HOME=/bazel_pip" -v "$(pwd)":/bazel_pip tensorflow/tensorflow:devel-py3 "./bazel_pip/tensorflow/tools/ci_build/builds/with_the_same_user" "./bazel_pip/tensorflow/tools/ci_build/builds/docker_cpu_pip.sh"
echo "This build not enabled for 2.0"
6 changes: 1 addition & 5 deletions tensorflow/tools/ci_build/release/windows/gpu_pip_on_cpu/build.bat
View file
Open in desktop
Expand Up @@ -13,8 +13,4 @@
:: limitations under the License.
:: =============================================================================

SET PYTHON_DIRECTORY=Python36

CALL tensorflow\tools\ci_build\release\common_win.bat

echo "This build does not exist in TF 2.0"
echo "This build not enabled on 2.0"