Anchor tf_cloud samples to 2.4.1 while investigating segfault

PiperOrigin-RevId: 379299277

Author: SinaChavoshi

g3doc/tutorials/distributed_training_nasnet_with_tensorflow_cloud.ipynb

@@ -392,8 +392,8 @@
         "\n",
         "# Optional: Some recommended base images. If you provide none the system\n",
         "# will choose one for you.\n",
-        "TF_GPU_IMAGE= \"tensorflow/tensorflow:latest-gpu\"\n",
-        "TF_CPU_IMAGE= \"tensorflow/tensorflow:latest\"\n",
+        "TF_GPU_IMAGE= \"tensorflow/tensorflow:2.4.1-gpu\"\n",
+        "TF_CPU_IMAGE= \"tensorflow/tensorflow:2.4.1\"\n",
         "\n",
         "# Submit a distributed training job using GPUs.\n",
         "tfc.run(\n",

g3doc/tutorials/hp_tuning_cifar10_using_google_cloud.ipynb

@@ -416,8 +416,8 @@
         "    f.write('keras-tuner\\n')\n",
         "\n",
         "# Optional: Some recommended base images. If you provide none the system will choose one for you.\n",
-        "TF_GPU_IMAGE= \"tensorflow/tensorflow:latest-gpu\"\n",
-        "TF_CPU_IMAGE= \"tensorflow/tensorflow:latest\"\n",
+        "TF_GPU_IMAGE= \"tensorflow/tensorflow:2.4.1-gpu\"\n",
+        "TF_CPU_IMAGE= \"tensorflow/tensorflow:2.4.1\"\n",
         "\n",
         "\n",
         "tfc.run_cloudtuner(\n",

g3doc/tutorials/hp_tuning_wide_and_deep_model.ipynb

@@ -550,9 +550,16 @@
       },
       "outputs": [],
       "source": [
+        "\n",
+        "# Optional: Some recommended base images. If you provide none the system will choose one for you.\n",
+        "TF_GPU_IMAGE= \"tensorflow/tensorflow:2.4.1-gpu\"\n",
+        "TF_CPU_IMAGE= \"tensorflow/tensorflow:2.4.1\"\n",
+        "\n",
+        "\n",
         "tfc.run_cloudtuner(\n",
         "    distribution_strategy='auto',\n",
         "    docker_config=tfc.DockerConfig(\n",
+        "        parent_image=TF_GPU_IMAGE,\n",
         "        image_build_bucket=GCS_BUCKET\n",
         "        ),\n",
         "    chief_config=tfc.MachineConfig(\n",

g3doc/tutorials/overview.ipynb

@@ -418,8 +418,8 @@
         "\n",
         "# Optional: Some recommended base images. If you provide none the system\n",
         "# will choose one for you.\n",
-        "TF_GPU_IMAGE= \"tensorflow/tensorflow:latest-gpu\"\n",
-        "TF_CPU_IMAGE= \"tensorflow/tensorflow:latest\"\n",
+        "TF_GPU_IMAGE= \"tensorflow/tensorflow:2.4.1-gpu\"\n",
+        "TF_CPU_IMAGE= \"tensorflow/tensorflow:2.4.1\"\n",
         "\n",
         "# Submit a single node training job using GPU.\n",
         "tfc.run(\n",