NOTE: Please rotate the CUDA repo keys as mentioned here: https://forums.developer.nvidia.com/t/notice-cuda-linux-repository-key-rotation/212772 and pull the container again from NGC: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/tensorflow

TensorFlow is an open source software library for numerical computation using data flow graphs. TensorFlow was originally developed by researchers and engineers for the purposes of conducting machine learning and deep neural networks research.

This image bundles NVIDIA's GPU-optimized TensorFlow container along with the base NGC AMI. The NGC AMI is an optimized environment for running the containers available on the NGC container registry. The Docker containers available on the NGC container registry are tuned, tested, and certified by NVIDIA to take full advantage of NVIDIA GPU's, the driving force behind innovations in artificial intelligence.

This image allows AWS users to immediately start using a GPU-optimized distribution of TensorFlow, greatly reducing the complexity required to set up an optimized software stack.

The 22.03.0 release contains:
Ubuntu Server 20.04
NVIDIA Driver 470.103.01
Docker-ce 20.10.12
NVIDIA Container Toolkit 1.8.1
NVIDIA Container Runtime 3.8.1
NVIDIA's distribution of TensorFlow 1 and 2, tags 22.02-tf2-py3 and 22.02-tf1-py3