Skip to content

Commit 10eef0f

Browse files
Merge pull request #2303 from 8bitmp3:update-tpu
PiperOrigin-RevId: 627147864
2 parents 9efadf0 + 0aeef4b commit 10eef0f

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

site/en/guide/tpu.ipynb

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
"id": "Tce3stUlHN0L"
77
},
88
"source": [
9-
"##### Copyright 2018 The TensorFlow Authors.\n"
9+
"##### Copyright 2024 The TensorFlow Authors.\n"
1010
]
1111
},
1212
{
@@ -81,7 +81,7 @@
8181
"id": "ebf7f8489bb7"
8282
},
8383
"source": [
84-
"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",
84+
"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",
8585
"\n",
8686
"Import some necessary libraries, including TensorFlow Datasets:"
8787
]
@@ -128,7 +128,7 @@
128128
},
129129
"outputs": [],
130130
"source": [
131-
"resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')\n",
131+
"resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='local')\n",
132132
"tf.config.experimental_connect_to_cluster(resolver)\n",
133133
"# This is the TPU initialization code that has to be at the beginning.\n",
134134
"tf.tpu.experimental.initialize_tpu_system(resolver)\n",
@@ -588,7 +588,6 @@
588588
"metadata": {
589589
"accelerator": "TPU",
590590
"colab": {
591-
"collapsed_sections": [],
592591
"name": "tpu.ipynb",
593592
"toc_visible": true
594593
},

0 commit comments

Comments
 (0)