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

Kaggle · Aug 17, 2023 · 226ac68 · 226ac68
1 parent 213a24a
commit 226ac68
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/Dockerfile.tmpl b/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 \