diff --git a/docs/applications/implementations/models.md b/docs/applications/implementations/models.md index 4262c0edb4..afa601407f 100644 --- a/docs/applications/implementations/models.md +++ b/docs/applications/implementations/models.md @@ -53,7 +53,7 @@ You can import PyPI packages or your own Python packages to help create more com The following packages have been pre-installed and can be used in your implementations: ```text -tensorflow==1.12.0 +tensorflow==1.13.1 boto3==1.9.78 msgpack==0.6.1 numpy>=1.13.3,<2 diff --git a/examples/mnist/requirements.txt b/examples/mnist/requirements.txt index cfbf63288b..79b8e2944b 100644 --- a/examples/mnist/requirements.txt +++ b/examples/mnist/requirements.txt @@ -1,2 +1,2 @@ pillow==5.4.1 -tensor2tensor==1.10.0 +tensor2tensor==1.13.4 diff --git a/examples/reviews/requirements.txt b/examples/reviews/requirements.txt index 267b3a6d6a..92c6587047 100644 --- a/examples/reviews/requirements.txt +++ b/examples/reviews/requirements.txt @@ -1 +1 @@ -tensor2tensor==1.10.0 +tensor2tensor==1.13.4 diff --git a/images/spark-base/Dockerfile b/images/spark-base/Dockerfile index 29873660b4..558a10d81f 100644 --- a/images/spark-base/Dockerfile +++ b/images/spark-base/Dockerfile @@ -12,7 +12,7 @@ RUN mkdir -p /opt ARG HADOOP_VERSION="2.9.2" ARG SPARK_VERSION="2.4.2" -ARG TF_VERSION="1.12.0" +ARG TF_VERSION="1.13.1" # Required for building tensorflow spark connector ARG SCALA_VERSION="2.12" # Scalatest version from https://github.com/apache/spark/blob/v2.4.2/pom.xml diff --git a/images/tf-base-gpu/Dockerfile b/images/tf-base-gpu/Dockerfile index d696be771a..85a6c26787 100644 --- a/images/tf-base-gpu/Dockerfile +++ b/images/tf-base-gpu/Dockerfile @@ -1,4 +1,4 @@ -FROM tensorflow/tensorflow:1.12.0-gpu-py3 +FROM tensorflow/tensorflow:1.13.1-gpu-py3 RUN apt-get update -qq && apt-get install -y -q \ zlib1g-dev \ diff --git a/images/tf-base/Dockerfile b/images/tf-base/Dockerfile index bde3c5daa2..5f4da35c22 100644 --- a/images/tf-base/Dockerfile +++ b/images/tf-base/Dockerfile @@ -1,4 +1,4 @@ -FROM tensorflow/tensorflow:1.12.0-py3 +FROM tensorflow/tensorflow:1.13.1-py3 RUN apt-get update -qq && apt-get install -y -q \ zlib1g-dev \ diff --git a/images/tf-serve-gpu/Dockerfile b/images/tf-serve-gpu/Dockerfile index de25316a76..780ed9925e 100644 --- a/images/tf-serve-gpu/Dockerfile +++ b/images/tf-serve-gpu/Dockerfile @@ -1,8 +1,8 @@ FROM cortexlabs/tf-base-gpu -ARG TF_VERSION="1.12.0" +ARG TF_SERV_VERSION="1.13.0" -RUN curl -o tensorflow-model-server.deb http://storage.googleapis.com/tensorflow-serving-apt/pool/tensorflow-model-server-${TF_VERSION}/t/tensorflow-model-server/tensorflow-model-server_${TF_VERSION}_all.deb +RUN curl -o tensorflow-model-server.deb http://storage.googleapis.com/tensorflow-serving-apt/pool/tensorflow-model-server-${TF_SERV_VERSION}/t/tensorflow-model-server/tensorflow-model-server_${TF_SERV_VERSION}_all.deb RUN dpkg -i tensorflow-model-server.deb ENTRYPOINT ["tensorflow_model_server"] diff --git a/images/tf-serve/Dockerfile b/images/tf-serve/Dockerfile index 7eedaaad5b..e46a88b52e 100644 --- a/images/tf-serve/Dockerfile +++ b/images/tf-serve/Dockerfile @@ -1,12 +1,12 @@ FROM cortexlabs/tf-base -ARG TF_VERSION="1.12.0" +ARG TF_SERV_VERSION="1.13.0" RUN apt-get update -qq && apt-get install -y -q \ curl \ && apt-get clean -qq && rm -rf /var/lib/apt/lists/* -RUN curl -o tensorflow-model-server.deb http://storage.googleapis.com/tensorflow-serving-apt/pool/tensorflow-model-server-${TF_VERSION}/t/tensorflow-model-server/tensorflow-model-server_${TF_VERSION}_all.deb +RUN curl -o tensorflow-model-server.deb http://storage.googleapis.com/tensorflow-serving-apt/pool/tensorflow-model-server-${TF_SERV_VERSION}/t/tensorflow-model-server/tensorflow-model-server_${TF_SERV_VERSION}_all.deb RUN dpkg -i tensorflow-model-server.deb ENTRYPOINT ["tensorflow_model_server"] diff --git a/pkg/workloads/lib/package.py b/pkg/workloads/lib/package.py index c5eec7fca0..f32e8d3996 100644 --- a/pkg/workloads/lib/package.py +++ b/pkg/workloads/lib/package.py @@ -39,7 +39,7 @@ def get_build_order(python_packages): def get_restricted_packages(): - req_list = ["pyspark==2.4.2", "tensorflow==1.12.0"] + req_list = ["pyspark==2.4.2", "tensorflow==1.13.1"] req_files = glob.glob("/src/**/requirements.txt", recursive=True) for req_file in req_files: