Add Ubuntu 16 gpu py38 nightly and release builds.
PiperOrigin-RevId: 296532794 Change-Id: I466c2c579a73390e8de1f69fadf7d3442d169e55
This commit is contained in:
Amit Patankar
TensorFlower Gardener
parent
f175cee3b6
commit
4e3e368ae6
@ -334,6 +334,9 @@ tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/nonpip.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/nonpip_v1.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/pip.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/pip_v1.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/gpu_py38_full/nightly_release.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/gpu_py38_full/nonpip.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/gpu_py38_full/pip.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/sanity/build.sh
|
||||
tensorflow/tools/ci_build/release/ubuntu_16/tpu_py37_full/nonpip.sh
|
||||
tensorflow/tools/ci_build/release/windows/cpu_libtensorflow/nightly.bat
|
||||
|
||||
@ -0,0 +1,78 @@
|
||||
#!/bin/bash
|
||||
# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
set -e
|
||||
set -x
|
||||
|
||||
source tensorflow/tools/ci_build/release/common.sh
|
||||
|
||||
set_bazel_outdir
|
||||
|
||||
install_ubuntu_16_pip_deps pip3.8
|
||||
|
||||
pip3.7 install --upgrade auditwheel --user
|
||||
|
||||
update_bazel_linux
|
||||
|
||||
python2.7 tensorflow/tools/ci_build/update_version.py --nightly
|
||||
|
||||
# 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_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
|
||||
export TF_NEED_TENSORRT=1
|
||||
export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
|
||||
export CC_OPT_FLAGS='-mavx'
|
||||
export PYTHON_BIN_PATH=$(which python3.8)
|
||||
yes "" | "$PYTHON_BIN_PATH" configure.py
|
||||
|
||||
# Build the pip package
|
||||
bazel build --config=opt --config=v2 \
|
||||
--crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1:toolchain \
|
||||
tensorflow/tools/pip_package:build_pip_package
|
||||
|
||||
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --nightly_flag
|
||||
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
|
||||
|
||||
# Upload the built packages to pypi.
|
||||
for WHL_PATH in $(ls pip_pkg/tf_nightly*dev*.whl); do
|
||||
|
||||
WHL_DIR=$(dirname "${WHL_PATH}")
|
||||
WHL_BASE_NAME=$(basename "${WHL_PATH}")
|
||||
AUDITED_WHL_NAME="${WHL_DIR}"/$(echo "${WHL_BASE_NAME//linux/manylinux2010}")
|
||||
|
||||
# Copy and rename for gpu manylinux as we do not want auditwheel to package in libcudart.so
|
||||
WHL_PATH=${AUDITED_WHL_NAME}
|
||||
cp "${WHL_DIR}"/"${WHL_BASE_NAME}" "${WHL_PATH}"
|
||||
echo "Copied manylinux2010 wheel file at: ${WHL_PATH}"
|
||||
|
||||
# test the whl pip package
|
||||
chmod +x tensorflow/tools/ci_build/builds/nightly_release_smoke_test.sh
|
||||
./tensorflow/tools/ci_build/builds/nightly_release_smoke_test.sh ${AUDITED_WHL_NAME}
|
||||
RETVAL=$?
|
||||
|
||||
# Upload the PIP package if whl test passes.
|
||||
if [ ${RETVAL} -eq 0 ]; then
|
||||
echo "Basic PIP test PASSED, Uploading package: ${AUDITED_WHL_NAME}"
|
||||
twine upload -r pypi-warehouse "${AUDITED_WHL_NAME}" || echo
|
||||
else
|
||||
echo "Basic PIP test FAILED, will not upload ${AUDITED_WHL_NAME} package"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
@ -0,0 +1,58 @@
|
||||
#!/bin/bash
|
||||
# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
set -e
|
||||
set -x
|
||||
|
||||
source tensorflow/tools/ci_build/release/common.sh
|
||||
|
||||
install_ubuntu_16_pip_deps pip3.8
|
||||
# Update bazel
|
||||
update_bazel_linux
|
||||
|
||||
# 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.8)
|
||||
export TF2_BEHAVIOR=1
|
||||
export PROJECT_NAME="tensorflow_gpu"
|
||||
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
|
||||
|
||||
# Get the default test targets for bazel.
|
||||
source tensorflow/tools/ci_build/build_scripts/PRESUBMIT_BUILD_TARGETS.sh
|
||||
|
||||
tag_filters="gpu,requires-gpu,-no_gpu,-no_oss,-oss_serial,-no_oss_py38"
|
||||
|
||||
bazel test --config=cuda --config=opt \
|
||||
--crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1:toolchain \
|
||||
--linkopt=-lrt \
|
||||
--action_env=TF2_BEHAVIOR="${TF2_BEHAVIOR}" \
|
||||
--test_lang_filters=py \
|
||||
--build_tag_filters=${tag_filters} \
|
||||
--test_tag_filters=${tag_filters} \
|
||||
--test_timeout="300,450,1200,3600" --local_test_jobs=4 \
|
||||
--test_output=errors --verbose_failures=true --keep_going \
|
||||
--run_under=//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute \
|
||||
-- ${DEFAULT_BAZEL_TARGETS} -//tensorflow/lite/...
|
||||
@ -0,0 +1,69 @@
|
||||
#!/bin/bash
|
||||
# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
set -e
|
||||
set -x
|
||||
|
||||
source tensorflow/tools/ci_build/release/common.sh
|
||||
|
||||
install_ubuntu_16_pip_deps pip3.8
|
||||
# Update bazel
|
||||
update_bazel_linux
|
||||
|
||||
# Export required variables for running pip.sh
|
||||
export OS_TYPE="UBUNTU"
|
||||
export CONTAINER_TYPE="GPU"
|
||||
export TF_PYTHON_VERSION='python3.8'
|
||||
|
||||
# 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 ${TF_PYTHON_VERSION})
|
||||
export PROJECT_NAME="tensorflow_gpu"
|
||||
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
|
||||
|
||||
# Get the default test targets for bazel.
|
||||
source tensorflow/tools/ci_build/build_scripts/PRESUBMIT_BUILD_TARGETS.sh
|
||||
|
||||
# Export optional variables for running pip.sh
|
||||
export TF_TEST_FILTER_TAGS='gpu,requires-gpu,-no_gpu,-no_oss,-oss_serial,-no_oss_py38'
|
||||
export TF_BUILD_FLAGS="--config=opt --config=v2 --config=cuda --distinct_host_configuration=false \
|
||||
--action_env=TF_CUDA_VERSION --action_env=TF_CUDNN_VERSION --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1:toolchain "
|
||||
export TF_TEST_FLAGS="--test_tag_filters=${TF_TEST_FILTER_TAGS} --build_tag_filters=${TF_TEST_FILTER_TAGS} \
|
||||
--distinct_host_configuration=false \
|
||||
--action_env=TF_CUDA_VERSION --action_env=TF_CUDNN_VERSION --test_env=TF2_BEHAVIOR=1 \
|
||||
--config=cuda --test_output=errors --local_test_jobs=4 --test_lang_filters=py \
|
||||
--verbose_failures=true --keep_going --define=no_tensorflow_py_deps=true \
|
||||
--run_under=//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute "
|
||||
export TF_TEST_TARGETS="${DEFAULT_BAZEL_TARGETS} -//tensorflow/lite/... "
|
||||
export TF_PIP_TESTS="test_pip_virtualenv_non_clean test_pip_virtualenv_clean"
|
||||
export IS_NIGHTLY=0 # Not nightly
|
||||
export TF_PROJECT_NAME=${PROJECT_NAME}
|
||||
export TF_PIP_TEST_ROOT="pip_test"
|
||||
|
||||
# To build both tensorflow and tensorflow-gpu pip packages
|
||||
export TF_BUILD_BOTH_GPU_PACKAGES=1
|
||||
|
||||
./tensorflow/tools/ci_build/builds/pip_new.sh
|
||||
Loading…
x
Reference in New Issue
Block a user