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

PyTorch from NVIDIA AMI

By: NVIDIA Latest Version: 20.08.1

Product Overview

NVDIA's GPU-optimized distribution of PyTorch.
PyTorch is a GPU accelerated tensor computational framework with a Python front end. Functionality can be easily extended with common Python libraries such as NumPy, SciPy and Cython.

This image bundles NVIDIA's container for PyTorch into the NGC base image for AWS. The NGC base image is an optimized environment for running the GPU optimized containers for Deep Learning and HPC available from the NGC (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.

This image allows user to immediately start using PyTorch . You can learn more about the PyTorch distribution from NVIDIA and look for newer releases of NVIDIA's PyTorch container at

This image release contains:
Ubuntu Server 18.04
NVIDIA Driver 450.51.06
Docker-ce 19.03.12
NVIDIA Container Toolkit 1.2.1-1
NVIDIA's GPU-optimized PyTorch container 20.07-py3





Operating System

Linux/Unix, Ubuntu Ubuntu 18.04

Delivery Methods

  • Amazon Machine Image

Pricing Information

Usage Information

Support Information

Customer Reviews