Upgrade TensorFlow version in all Dockerfiles to v2

Signed-off-by: Sandipan Panda <[email protected]>
kubeflow · Mar 10, 2024 · 995048a · 995048a
1 parent 14eeaeb
commit 995048a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/examples/tensorflow/dist-mnist/Dockerfile b/examples/tensorflow/dist-mnist/Dockerfile
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM tensorflow/tensorflow:1.5.0
+FROM tensorflow/tensorflow:2.16.1
 
 ADD examples/tensorflow/dist-mnist/ /var/tf_dist_mnist
 ENTRYPOINT ["python", "/var/tf_dist_mnist/dist_mnist.py"]
diff --git a/examples/tensorflow/distribution_strategy/estimator-API/Dockerfile b/examples/tensorflow/distribution_strategy/estimator-API/Dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:1.11.0
+FROM tensorflow/tensorflow:2.16.1
 
 COPY examples/tensorflow/distribution_strategy/estimator-API/keras_model_to_estimator.py /
 ENTRYPOINT ["python", "/keras_model_to_estimator.py", "/tmp/tfkeras_example/"]
diff --git a/examples/tensorflow/mnist_with_summaries/Dockerfile b/examples/tensorflow/mnist_with_summaries/Dockerfile
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM tensorflow/tensorflow:1.11.0
+FROM tensorflow/tensorflow:2.16.1
 
 ADD examples/tensorflow/mnist_with_summaries/ /var/tf_mnist
 ENTRYPOINT ["python", "/var/tf_mnist/mnist_with_summaries.py"]
diff --git a/examples/tensorflow/tf_sample/Dockerfile b/examples/tensorflow/tf_sample/Dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:1.8.0
+FROM tensorflow/tensorflow:2.16.1
 RUN pip install retrying
 RUN mkdir -p /opt/kubeflow
 COPY examples/tensorflow/tf_sample/tf_smoke.py /opt/kubeflow/