Update tfjob

polyaxon · Jun 23, 2020 · 5c09a6f · 5c09a6f
1 parent 36888cc
commit 5c09a6f
Show file tree

Hide file tree

Showing 20 changed files with 460 additions and 1,344 deletions.
diff --git a/in_cluster/build_image/build.yaml b/in_cluster/build_image/build.yaml
@@ -9,11 +9,13 @@ params:
 runPatch:
   init:
   - dockerfile:
-      image: "tensorflow/tensorflow:2.0.1-py3"
+      image: "tensorflow/tensorflow:2.2.0"
       run:
       - 'pip3 install --no-cache-dir -U polyaxon["s3","gcs","azure","polyboard","polytune"]'
       - pip3 install mxnet
       - pip3 install scikit-learn
       - pip3 install xgboost
+      - pip3 install keras
+      - pip3 install tensorflow_datasets
       langEnv: 'en_US.UTF-8'
 hubRef: kaniko
diff --git a/in_cluster/horovod/keras/Dockerfile b/in_cluster/horovod/keras/Dockerfile
diff --git a/in_cluster/horovod/keras/__init__.py b/in_cluster/horovod/keras/__init__.py
diff --git a/in_cluster/horovod/keras/mnist.py b/in_cluster/horovod/keras/mnist.py
diff --git a/in_cluster/horovod/mxnet/Dockerfile b/in_cluster/horovod/mxnet/Dockerfile
diff --git a/in_cluster/horovod/mxnet/__init__.py b/in_cluster/horovod/mxnet/__init__.py