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

PyTorch 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 from NGC:

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 23.03.0 image 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 GPU-optimized PyTorch container 22.02-py3





Operating System

Linux/Unix, Ubuntu Ubuntu 20.04

Delivery Methods

  • Amazon Machine Image

Pricing Information

Usage Information

Support Information

Customer Reviews