Updated TFMA and TFDV versions of Jupyter images (kubeflow#2666)

* updated TFMA and TFDV versions of Jupyter images * applied updates on gpu images * fixed the version numbers
StatCan · Mar 11, 2019 · 5c6744f · 5c6744f
1 parent b82e206
commit 5c6744f
Showing 4 changed files with 11 additions and 4 deletions.
diff --git a/components/tensorflow-notebook-image/versions/1.11.0/version-config.json b/components/tensorflow-notebook-image/versions/1.11.0/version-config.json
@@ -2,5 +2,6 @@
   "BASE_IMAGE": "ubuntu:18.04@sha256:de774a3145f7ca4f0bd144c7d4ffb2931e06634f11529653b23eba85aef8e378",
   "TF_PACKAGE": "https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.11.0-cp36-cp36m-linux_x86_64.whl",
   "TF_PACKAGE_PY_27": "https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.11.0-cp27-none-linux_x86_64.whl",
-  "INSTALL_TFMA": "yes"
+  "TFMA_VERSION": "0.11.0",
+  "TFDV_VERSION": "0.11.0"
 }
diff --git a/components/tensorflow-notebook-image/versions/1.11.0gpu/version-config.json b/components/tensorflow-notebook-image/versions/1.11.0gpu/version-config.json
@@ -2,5 +2,7 @@
   "BASE_IMAGE": "nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04",
   "TF_PACKAGE": "https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.11.0-cp36-cp36m-linux_x86_64.whl",
   "TF_PACKAGE_PY_27": "https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.11.0-cp27-none-linux_x86_64.whl",
-  "INSTALL_TFMA": "yes"
+  "TFMA_VERSION": "0.11.0",
+  "TFDV_VERSION": "0.11.0"
 }
+
diff --git a/components/tensorflow-notebook-image/versions/1.12.0/version-config.json b/components/tensorflow-notebook-image/versions/1.12.0/version-config.json
@@ -2,5 +2,7 @@
   "BASE_IMAGE": "ubuntu:18.04@sha256:de774a3145f7ca4f0bd144c7d4ffb2931e06634f11529653b23eba85aef8e378",
   "TF_PACKAGE": "https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.12.0-cp36-cp36m-linux_x86_64.whl",
   "TF_PACKAGE_PY_27": "https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.12.0-cp27-none-linux_x86_64.whl",
-  "INSTALL_TFMA": "yes"
+  "TFMA_VERSION": "0.12.0",
+  "TFDV_VERSION": "0.12.0"
 }
+
diff --git a/components/tensorflow-notebook-image/versions/1.12.0gpu/version-config.json b/components/tensorflow-notebook-image/versions/1.12.0gpu/version-config.json
@@ -2,5 +2,7 @@
   "BASE_IMAGE": "nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04",
   "TF_PACKAGE": "https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.12.0-cp36-cp36m-linux_x86_64.whl",
   "TF_PACKAGE_PY_27": "https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.12.0-cp27-none-linux_x86_64.whl",
-  "INSTALL_TFMA": "yes"
+  "TFMA_VERSION": "0.12.0",
+  "TFDV_VERSION": "0.12.0"
 }
+