This page helps you to choose which container image you want to use.
Choose a container image type
Each container image provides a Python 3 environment and includes the
selected data science framework (such as PyTorch or TensorFlow), Conda, the
NVIDIA stack for GPU images (CUDA, cuDNN, NCCL2), and many other
supporting packages and tools. To find the container image that you want,
see the table below.
The following list of Deep Learning Containers image types is
organized by framework type.
TensorFlow Enterprise container images provide you with a Google Cloud
optimized distribution of TensorFlow, and specific versions of
the TensorFlow Enterprise distribution also include Long Term Version
To learn more about TensorFlow Enterprise, read the
TensorFlow Enterprise overview.
Some Deep Learning Containers 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.
Listing all available versions
If you need a specific framework or CUDA version, you can search
the complete list of available container images. To list all available
Deep Learning Containers images, use the following command in
the gcloud command-line tool with your preferred terminal or in
gcloud container images list --repository="gcr.io/deeplearning-platform-release"