experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Only update default docker images for official TF release versions.

Browse Source 
PiperOrigin-RevId: 291249321
Change-Id: I7676cde9b79c22702b21c9d7bfd8ca8f64f721a7
This commit is contained in:
Yifei Feng 2020-01-23 14:47:15 -08:00 committed by TensorFlower Gardener
parent e4d2552523
commit 77bed6590d
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tensorflow/tools/ci_build/release/ubuntu_16/custom_op/release.sh
View File

@ -21,8 +21,6 @@ if [[ -z "${TF_VERSION}" ]]; then
exit
fi
UBUNTU16_CPU_IMAGE="tensorflow/tensorflow:custom-op-ubuntu16"
UBUNTU16_GPU_IMAGE="tensorflow/tensorflow:custom-op-gpu-ubuntu16"
VERSIONED_UBUNTU16_CPU_IMAGE="tensorflow/tensorflow:${TF_VERSION}-custom-op-ubuntu16"
VERSIONED_UBUNTU16_GPU_IMAGE="tensorflow/tensorflow:${TF_VERSION}-custom-op-gpu-ubuntu16"
@ -37,11 +35,18 @@ docker login -u "${TF_DOCKER_USERNAME}" -p "${TF_DOCKER_PASSWORD}"
docker push "${VERSIONED_UBUNTU16_CPU_IMAGE}"
docker push "${VERSIONED_UBUNTU16_GPU_IMAGE}"
# Tag and push the default images
docker tag "${VERSIONED_UBUNTU16_CPU_IMAGE}" "${UBUNTU16_CPU_IMAGE}"
docker push "${UBUNTU16_CPU_IMAGE}"
# Tag and push the default images for official TF releases
if [[ ${TF_VERSION} == *"rc"* ]]; then
echo "Do not update default images as ${TF_VERSION} is a release candidate."
else
UBUNTU16_CPU_IMAGE="tensorflow/tensorflow:custom-op-ubuntu16"
UBUNTU16_GPU_IMAGE="tensorflow/tensorflow:custom-op-gpu-ubuntu16"
docker tag "${VERSIONED_UBUNTU16_GPU_IMAGE}" "${UBUNTU16_GPU_IMAGE}"
docker push "${UBUNTU16_GPU_IMAGE}"
docker tag "${VERSIONED_UBUNTU16_CPU_IMAGE}" "${UBUNTU16_CPU_IMAGE}"
docker push "${UBUNTU16_CPU_IMAGE}"
docker tag "${VERSIONED_UBUNTU16_GPU_IMAGE}" "${UBUNTU16_GPU_IMAGE}"
docker push "${UBUNTU16_GPU_IMAGE}"
fi
docker logout#!/bin/bash