Change H2O to resolve via conda and bundle tf into its own layer

Author: neshdev

Dockerfile.tmpl

@@ -183,10 +183,8 @@ RUN JAXVER=$(pip freeze | grep -e "^jax==") && \
         flax \
         "${JAXVER}" && \
 
-    # Install h2o from source.
-    # Use `conda install -c h2oai h2o` once Python 3.7 version is released to conda.
-    apt-get install -y default-jre-headless && \
-    pip install -f https://h2o-release.s3.amazonaws.com/h2o/latest_stable_Py.html h2o \
+RUN conda install -c h2oai h2o && /tmp/clean-layer.sh
+RUN pip install \
         "tensorflow-gcs-config<=${TENSORFLOW_VERSION}" \
         "tensorflow==${TENSORFLOW_VERSION}" \
         tensorflow-addons \