Merge pull request #2303 from 8bitmp3:update-tpu

PiperOrigin-RevId: 627147864
tensorflow · Apr 22, 2024 · 10eef0f · 10eef0f
2 parents 9efadf0 + 0aeef4b
commit 10eef0f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/site/en/guide/tpu.ipynb b/site/en/guide/tpu.ipynb
@@ -6,7 +6,7 @@
         "id": "Tce3stUlHN0L"
       },
       "source": [
-        "##### Copyright 2018 The TensorFlow Authors.\n"
+        "##### Copyright 2024 The TensorFlow Authors.\n"
       ]
     },
     {
@@ -81,7 +81,7 @@
         "id": "ebf7f8489bb7"
       },
       "source": [
-        "Before you run this Colab notebook, make sure that your hardware accelerator is a TPU by checking your notebook settings: **Runtime** > **Change runtime type** > **Hardware accelerator** > **TPU**.\n",
+        "Before you run this Colab notebook, make sure that your hardware accelerator is a TPU by checking your notebook settings: **Runtime** > **Change runtime type** > **Hardware accelerator** > **TPU v2**.\n",
         "\n",
         "Import some necessary libraries, including TensorFlow Datasets:"
       ]
@@ -128,7 +128,7 @@
       },
       "outputs": [],
       "source": [
-        "resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')\n",
+        "resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='local')\n",
         "tf.config.experimental_connect_to_cluster(resolver)\n",
         "# This is the TPU initialization code that has to be at the beginning.\n",
         "tf.tpu.experimental.initialize_tpu_system(resolver)\n",
@@ -588,7 +588,6 @@
   "metadata": {
     "accelerator": "TPU",
     "colab": {
-      "collapsed_sections": [],
       "name": "tpu.ipynb",
       "toc_visible": true
     },