Using TensorFlow on Google Colab

Follow the next set of steps to get started with using TensorFlow on Google Colab. Note that these steps are specific to Google Colab only, where TensorFlow already comes preinstalled:

  1. First, visit the Google Colab URL, at https://colab.research.google.com.
  2. Create a new Python 3 Jupyter notebook:

  1. Change the runtime to the GPU as discussed previously in Chapter 9Containerization on GPU-Enabled Platforms (cloud containers):

  1. Confirm the changes by listing all the devices to get the GPU information with the ...

Get Hands-On GPU Computing with Python now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.