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
-notebook
in
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
|
TensorFlow 1.x | GPU | tf-1-13-cu100-notebooks tf-1-14-cu100-notebooks tf-1-15-cu110-notebooks
|
Swift for TensorFlow (experimental) | GPU | swift-0-11-cu110-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 with XLA (experimental) | XLA | pytorch-1-6-xla-notebooks
|
R (experimental) | CPU | r-3-5-cpu-experimental-notebooks r-3-6-cpu-experimental-notebooks
|
RAPIDS (experimental) | GPU | rapids-0-7-gpu-experimental-notebooks rapids-0-12-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.
Swift for TensorFlow is only available in Ubuntu 18.04.
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-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
Create a new AI Platform Notebooks instance using the Google Cloud Console or using the command line.
Learn more about Deep Learning VM instances in the Deep Learning VM documentation.