Skip to content

Update to Tensorflow 1.13.1 #116

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.

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

Merged
merged 3 commits into from
Jun 4, 2019
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/applications/implementations/models.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion examples/mnist/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
pillow==5.4.1
tensor2tensor==1.10.0
tensor2tensor==1.13.4
2 changes: 1 addition & 1 deletion examples/reviews/requirements.txt
Original file line number Diff line number Diff line change
@@ -1 +1 @@
tensor2tensor==1.10.0
tensor2tensor==1.13.4
2 changes: 1 addition & 1 deletion images/spark-base/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion images/tf-base-gpu/Dockerfile
Original file line number Diff line number Diff line change
@@ -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 \
Expand Down
2 changes: 1 addition & 1 deletion images/tf-base/Dockerfile
Original file line number Diff line number Diff line change
@@ -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 \
Expand Down
4 changes: 2 additions & 2 deletions images/tf-serve-gpu/Dockerfile
Original file line number Diff line number Diff line change
@@ -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"]
4 changes: 2 additions & 2 deletions images/tf-serve/Dockerfile
Original file line number Diff line number Diff line change
@@ -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"]
2 changes: 1 addition & 1 deletion pkg/workloads/lib/package.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down