"...entrypoints/git@developer.sourcefind.cn:norm/vllm.git" did not exist on "4298374265a4379a2bd378373c7252b7a7b2b34f"
Unverified Commit b1faa20b authored by Karmel Allison's avatar Karmel Allison Committed by GitHub
Browse files

Use nightly TensorFlow docker image for tests (#3256)

* Using nightly TF docker

* Explicitly pull image
parent d0aa6081
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# ============================================================================== # ==============================================================================
# #
# DO NOT MODIFY THIS FILE. Add tests to be executed in test_models.sh # DO NOT MODIFY THIS FILE. Add tests to be executed in test_models.sh
# Usage: docker_test.sh [--docker-image <DOCKER_IMG_NAME>] # Usage: docker_test.sh [--docker-image <DOCKER_IMG_NAME>]
# #
...@@ -22,7 +22,7 @@ ...@@ -22,7 +22,7 @@
# --docker-image flag), the default latest tensorflow docker # --docker-image flag), the default latest tensorflow docker
# will be used. # will be used.
# #
# The script obeys the following required environment variables unless superceded by # The script obeys the following required environment variables unless superceded by
# the docker image flag: # the docker image flag:
# PYTHON_VERSION: (PYTHON2 | PYTHON3) # PYTHON_VERSION: (PYTHON2 | PYTHON3)
...@@ -35,9 +35,9 @@ EXIT=0 ...@@ -35,9 +35,9 @@ EXIT=0
export WORKSPACE=${PWD} export WORKSPACE=${PWD}
if [ "$PYTHON_VERSION" = "PYTHON3" ]; then if [ "$PYTHON_VERSION" = "PYTHON3" ]; then
DOCKER_IMG_NAME="tensorflow/tensorflow:1.4.0-py3" DOCKER_IMG_NAME="tensorflow/tensorflow:nightly-py3"
else else
DOCKER_IMG_NAME="tensorflow/tensorflow:1.4.0" DOCKER_IMG_NAME="tensorflow/tensorflow:nightly"
if [ "$PYTHON_VERSION" != "PYTHON2" ]; then if [ "$PYTHON_VERSION" != "PYTHON2" ]; then
echo "WARNING: Python version was not specified. Using Python2 by default." echo "WARNING: Python version was not specified. Using Python2 by default."
sleep 5 sleep 5
...@@ -56,6 +56,9 @@ fi ...@@ -56,6 +56,9 @@ fi
# Specify which test is to be run # Specify which test is to be run
COMMAND="./official/testing/test_models.sh" COMMAND="./official/testing/test_models.sh"
# Check the recency of the desired image
${DOCKER_BINARY} pull ${DOCKER_IMG_NAME}
# RUN # RUN
${DOCKER_BINARY} run \ ${DOCKER_BINARY} run \
-v ${WORKSPACE}:/workspace \ -v ${WORKSPACE}:/workspace \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment