The NVIDIA Deep Learning AMI is an optimized environment for running the Deep Learning, Data Science, and HPC containers available from NVIDIA's NGC Catalog. The Docker containers available on the NGC Catalog are tuned, tested, and certified by NVIDIA to take full advantage of NVIDIA Ampere, Volta and Turing Tensor Cores, the driving force behind artificial intelligence. Deep Learning, Data Science, and HPC containers from the NGC Catalog require this AMI for the best GPU acceleration on AWS P4D, P3 and G4 instances.

NVIDIA Deep Learning AMI Release Version 21.02.2 includes:
Ubuntu Server 18.04
NVIDIA Driver 460.32.03
Docker-ce 19.03.14
NVIDIA Container Toolkit 1.4.1-1
NVIDIA Container Runtime 3.4.1-1

Read more at: http://docs.nvidia.com/ngc/ngc-ami-release-notes/