Choosing a virtual machine (VM) image

AI Platform Notebooks instances are AI Platform Deep Learning VM Image instances with JupyterLab notebook environments enabled and ready for use. Specific AI Platform Notebooks images are available to suit your choice of framework and processor. To find the image that you want, see the following table.

Deciding on an image family

Deciding on which AI Platform Notebooks image family to use depends on your needs. The following table lists the default versions of image families, organized by framework type. Creating an instance by referencing an image family with -notebookin the name ensures that your instance is a supported image family. If you need a specific framework version that is not shown here, skip to Listing all available versions.

Framework Processor Image family name(s)
Base GPU common-cu90-notebooks
common-cu91-notebooks
common-cu92-notebooks
common-cu100-notebooks
common-cu101-notebooks
common-cu110-notebooks
CPU common-cpu-notebooks
TensorFlow Enterprise 2.x GPU tf2-ent-2-1-cu110-notebooks
tf2-ent-2-3-cu110-notebooks
TensorFlow Enterprise 1.x GPU tf-ent-1-15-cu110-notebooks
TensorFlow 2.x GPU tf2-2-0-cu100-notebooks
tf2-2-2-cu101-notebooks
tf2-2-4-cu110-notebooks
TensorFlow 1.x GPU tf-1-13-cu100-notebooks
tf-1-14-cu100-notebooks
PyTorch GPU pytorch-1-1-cu100-notebooks
pytorch-1-2-cu100-notebooks
pytorch-1-3-cu100-notebooks
pytorch-1-4-cu101-notebooks
pytorch-1-6-cu110-notebooks
pytorch-1-7-cu110-notebooks
pytorch-1-8-cu110-notebooks
PyTorch XLA TPU/GPU/CPU (experimental) pytorch-1-6-xla-notebooks
pytorch-1-7-xla-notebooks
pytorch-1-8-xla-notebooks
R CPU (experimental) r-3-5-cpu-experimental-notebooks
r-3-6-cpu-experimental-notebooks
r-4-0-cpu-experimental-notebooks
RAPIDS GPU (experimental) rapids-0-7-gpu-experimental-notebooks
rapids-0-12-gpu-experimental-notebooks
rapids-0-18-gpu-experimental-notebooks

Choosing an operating system

For most frameworks, Debian 10 is the default OS. Debian 9 and Ubuntu 18.04 images are available for some frameworks. They are denoted by the -debian-9 and -ubuntu-1804 suffixes in the image family name (see Listing all available versions). For the RAPIDS framework, Debian 9 is the default and it is not available in Debian 10.

PyTorch 1.6 Cuda 11 (pytorch-1-6-cu110-notebooks and all TensorFlow Enterprise (tf-ent and tf2-ent) image families support A100 GPU accelerators.

TensorFlow Enterprise images

TensorFlow Enterprise image families provide you with a Google Cloud optimized distribution of TensorFlow with Long Term Version Support. To learn more about TensorFlow Enterprise, read the TensorFlow Enterprise overview.

Use the following table of available TensorFlow images to help you select the image with the version of TensorFlow or TensorFlow Enterprise that you want.

Version of TensorFlow or TensorFlow Enterprise Processor Image family name
TensorFlow Enterprise 2.3 GPU tf2-ent-2-3-cu110-notebooks
TensorFlow 2.2 GPU tf2-2-2-cu101-notebooks
TensorFlow Enterprise 2.1 GPU tf2-ent-2-1-cu110-notebooks
TensorFlow 2.0 GPU tf2-2-0-cu100-notebooks
TensorFlow Enterprise 1.15 GPU tf-ent-1-15-cu110-notebooks
TensorFlow 1.14 GPU tf-1-14-cu100-notebooks
TensorFlow 1.13 GPU tf-1-13-cu100-notebooks

Experimental images

Some AI Platform Notebooks image families are experimental, as indicated by the table of image families. Experimental images are supported on a best-effort basis, and may not receive refreshes on each new release of the framework.

Specifying an image version

Creating a new AI Platform Notebooks instance based on the image family name gives you the most recent image of that version of the framework. For example, if you create an AI Platform Notebooks instance based on the family name tf-ent-1-15-cu110-notebooks, the specific image name might look like tf-ent-1-15-cu110-notebooks-v20201016.

To create multiple AI Platform Notebooks instances based on the exact same image, use the image name instead of the image family name.

To determine the exact name of the most recent image, use the following command in the gcloud command-line tool with your preferred terminal or in Cloud Shell. Replace IMAGE_FAMILY with the image family name for which you want to find out the most recent version number.

gcloud compute images describe-from-family IMAGE_FAMILY \
        --project deeplearning-platform-release

Look for the name field in the output and use the image name given there when creating new instances.

Listing all available versions

If you need a specific framework, CUDA version, or operating system, you can search the complete list of available images. To list all available AI Platform Notebooks images, use the following gcloud tool command.

gcloud compute images list \
        --project deeplearning-platform-release | grep notebooks

Image families will be in the format FRAMEWORK-VERSION-CUDA_VERSION(-experimental)-notebooks, where FRAMEWORK is the target library, VERSION is the framework version, and CUDA_VERSION is the version of the CUDA stack, if present.

For example, an image from the family tf2-ent-2-3-cu110-notebooks has TensorFlow Enterprise 2.3 and CUDA 11.0.

What's next