Removes a deprecation warning. Tested the updated .ipynb file.

PiperOrigin-RevId: 404370475
chunduriv · Oct 19, 2021 · 102913a · 102913a
1 parent 1fb76ca
commit 102913a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/tutorials/bert_glue.ipynb b/docs/tutorials/bert_glue.ipynb
@@ -227,7 +227,7 @@
         "  tf.tpu.experimental.initialize_tpu_system(cluster_resolver)\n",
         "  strategy = tf.distribute.TPUStrategy(cluster_resolver)\n",
         "  print('Using TPU')\n",
-        "elif tf.test.is_gpu_available():\n",
+        "elif tf.config.list_physical_devices('GPU'):\n",
         "  strategy = tf.distribute.MirroredStrategy()\n",
         "  print('Using GPU')\n",
         "else:\n",