Sign in
Your Saved List Partners Sell in AWS Marketplace Amazon Web Services Home Help

TensorFlow from NVIDIA AMI

By: NVIDIA Latest Version: 22.03.0

Product Overview

NOTE: Please rotate the CUDA repo keys as mentioned here: and pull the container again from NGC:

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





Operating System

Linux/Unix, Ubuntu Ubuntu 20.04

Delivery Methods

  • Amazon Machine Image

Pricing Information

Usage Information

Support Information

Customer Reviews