From e464668efc3e1ad96a277fe9da48fceffb605cd7 Mon Sep 17 00:00:00 2001 From: Nathan Roberts Date: Fri, 25 Feb 2022 09:41:57 +0000 Subject: [PATCH] Update deeplearning-platfrom-release image in Dockerfile Deprecated image was in suggested Dockerfile leading to this error: from tensorflow.contrib.boosted_trees.python.ops import gen_quantile_ops ModuleNotFoundError: No module named 'tensorflow.contrib' --- src/python/tensorflow_cloud/tuner/cloud_fit_readme.md | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/python/tensorflow_cloud/tuner/cloud_fit_readme.md b/src/python/tensorflow_cloud/tuner/cloud_fit_readme.md index 4362d44b..c8be5f1e 100644 --- a/src/python/tensorflow_cloud/tuner/cloud_fit_readme.md +++ b/src/python/tensorflow_cloud/tuner/cloud_fit_readme.md @@ -143,9 +143,10 @@ you can skip the setup and authentication steps and start from step 8. Create a dockerfile as follows: ```shell - # Using DLVM base image. For GPU training use - # gcr.io/deeplearning-platform-release/tf2-gpu instead. - FROM gcr.io/deeplearning-platform-release/tf2-cpu + # Using DLVM base image. + # List of images to choose from: https://cloud.google.com/deep-learning-containers/docs/choosing-container + # Check for correct image according to CPU / GPU need. + FROM gcr.io/deeplearning-platform-release/tf-cpu.2-8 WORKDIR /root # Path configuration