The NVIDIA GPU-Optimized AMI is an environment for running the GPU-accelerated deep learning and HPC containers from the NVIDIA NGC catalog. The deep learning containers from NGC catalog require this AMI for GPU acceleration on AWS P5d, P4d, P3, G4dn, G5 GPU instances.
The NVIDIA GPU-Optimized AMI is a virtual machine image for accelerating your GPU accelerated Machine Learning, Deep Learning, Data Science and HPC workloads. Using this AMI, you can spin up a GPU-accelerated EC2 VM instance in minutes with a pre-installed Ubuntu OS, GPU driver, Docker and NVIDIA container toolkit.
This AMI provides easy access to NVIDIA's NGC Catalog, a hub for GPU-optimized software, for pulling & and running performance-tuned, tested, and NVIDIA certified docker containers. The NGC catalog provides free access to containerized AI, Data Science, and HPC applications, pre-trained models, AI SDKs and other resources to enable data scientists, developers, and researchers to focus on building and deploying solutions.
This GPU-optimized AMI is free with an option to purchase enterprise support offered through NVIDIA AI Enterprise. For how to get support for this AMI, scroll down to 'Support Information'
NVIDIA GPU-Optimized AMI includes:
Ubuntu Server OS
NVIDIA Driver
Docker-ce
NVIDIA Container Toolkit
AWS CLI, NGC CLI
Miniconda, JupyterLab, Git
Highlights
Provides data scientists and developers fast and easy access to NVIDIA H100, A100, A10 and T4 GPUs in the cloud and GPU-optimized AI/HPC software in an environment that is fully certified by NVIDIA.
Optimized for highest performance across a wide range of workloads on NVIDIA GPUs
NVIDIA accelerates innovation by eliminating the complex do-it-yourself task of building and optimizing a complete deep learning software stack tuned specifically for GPUs.
AWS Marketplace now accepts line of credit payments through the PNC Vendor Finance program. This program is available to select AWS customers in the US, excluding NV, NC, ND, TN, & VT.
You pay nothing for this software; it is offered free of charge. The pricing table lists 20 EC2 instance options, all billed by the hour. These are not software tiers — each entry maps to a specific GPU instance size in the g4dn, g5, p3, p3dn, p4d, and p5 families. Your cost scales with the instance you choose: larger instances with more GPUs, CPU, and memory carry higher hourly rates. You select the instance that fits your GPU, CPU, and memory needs, then run it for the hours you use.
Top-of-mind questions for buyers
What GPU hardware do the different instance families map to?
Each instance name ties to a specific NVIDIA GPU generation. The g4dn family uses NVIDIA T4 GPUs. The p3 and p3dn families use NVIDIA V100 GPUs. The p4d family uses NVIDIA A100 GPUs. Larger instance sizes within a family add more GPUs, CPU, and memory.
The software is free, so am I charged anything when I run an instance?
The software carries no license charge. You still pay the underlying AWS EC2 compute rate for the instance you run, billed by the hour. Charges accrue while the instance runs. Stopping the instance ends compute charges, though attached storage may still incur separate AWS fees.
Does the free software include any support?
The GPU-optimized AMI is offered free of charge for developers, with an enterprise support option available separately. Base access gives you the NVIDIA Developer Forum community. Direct access to NVIDIA experts, upgrade control, and knowledge-base resources come through a separate enterprise support license, not this listing.
docs.nvidia.com+1
Helpful?
Vendor refund policy
This AMI is provided free of charge.
How can we make this page better?
Tell us how we can improve this page, or report an issue with this product.
Give us feedbackReport a problem with this product or seller
Legal
Vendor terms and conditions
Upon subscribing to this product, you must acknowledge and agree to the terms and conditions outlined in the vendor's End User License Agreement (EULA).
Content disclaimer
Vendors are responsible for their product descriptions and other product content. AWS does not warrant that vendors' product descriptions or other product content are accurate, complete, reliable, current, or error-free.
An AMI is a virtual image that provides the information required to launch an instance. Amazon EC2 (Elastic Compute Cloud) instances are virtual servers on which you can run your applications and workloads, offering varying combinations of CPU, memory, storage, and networking resources. You can launch as many instances from as many different AMIs as you need.
Version release notes
NVIDIA Enterprise AI VMI 2026.4.1 release with NVIDIA driver 595.58.03
AWS Support is a one-on-one, fast-response support channel that is staffed 24x7x365 with experienced and technical support engineers. The service helps customers of all sizes and technical abilities to successfully utilize the products and features provided by Amazon Web Services.
The NVIDIA Omniverse™ Development Workstation (Windows) provides a pre-configured environment containing the software and drivers required to accelerate Omniverse development.
NVIDIA RTX Virtual Workstation delivers unmatched NVIDIA RTX performance for AI-enhanced and graphics-intensive applications in the cloud, offering workstation-grade power. Key advantages include ISV certifications with the NVIDIA RTX platform, streamlined IT infrastructure management, exceptional scalability, and access to the latest NVIDIA vGPU drivers and security patches.
NVIDIA RTX Virtual Workstation delivers unmatched NVIDIA RTX performance for AI-enhanced and graphics-intensive applications in the cloud, offering workstation-grade power. Key advantages include ISV certifications with the NVIDIA RTX platform, streamlined IT infrastructure management, exceptional scalability, and access to the latest NVIDIA vGPU drivers and security patches.
You need to first change the username from root to ubuntu in order to have the drivers be installed! I feel this should have been more specified in the directions!
yikesawjeez
AMI is not configured as advertised.
Reviewed on Mar 15, 2024
Review from a verified AWS customer
None of the advertised utilities are installed in the AMI, neither is CUDA. This is current as of 3/14/24. It appears to be a raw installation of 22.04, by my estimation.
root@ip-172-31-38-109:/cuda-samples/Samples/5_Domain_Specific/nbody# jupyterlab --version jupyterlab: command not found root@ip-172-31-38-109:/cuda-samples/Samples/5_Domain_Specific/nbody# miniconda --version miniconda: command not found
There's been a lot of troubleshooting so far with regard to attempting to get cuda installed, so I won't copy-paste my terminal.
Dan
Drives auto installed on login not boot
Reviewed on Feb 15, 2024
Review from a verified AWS customer
I wanted to use this AMI in my automation to run ML jobs in our platform. What I needed was a Ubuntu 22.04, because podman is in the repo, and Nvidia drivers installed. The downside of this AMI is, Nvidia drivers are installed via /home/ubuntu/.bashrc and not cloud-init. I looked at /var/tmp/nvidia/driver.sh and there was no variable to set to force driver install at cloud-init. Since my automation runs at the end of cloud-init this doesn't work.
Ema
Very good
Reviewed on Jan 20, 2024
Review from a verified AWS customer
Older reviews are not valid anymore, now at the date of my review the image is very good, it has all the drivers required to run optimized code on various types of NVIDIA GPUs, it has CUDA 12.1 preinstalled and also miniconda and Jupyterlab. The machine is ready to run code on GPU very easily with everything you need already in place.
AI researcher unhappy with NVIDIA software
Missing drivers
Reviewed on Dec 18, 2023
Review from a verified AWS customer
This should be preconfigured to run NVIDIA GPU Cloud (NGC) containers such as the PyTorch one, however it fails on launch on AWS (on a p3.2xlarge instance).
After sshing in, I see this error message: <br/>Installing drivers ...<br/>modprobe: FATAL: Module nvidia not found in directory /lib/modules/6.2.0-1011-aws<br/> And sure enough, running containers such as PyTorch (https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch) does not work:
<br/>~$ docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:23.11-py3<br/>docker: Error response from daemon: failed to create task for container: failed to create shim task: OCI runtime create failed: runc create failed: unable to start container process: error during container init: error running hook #0: error running hook: exit status 1, stdout: , stderr: Auto-detected mode as 'legacy'<br/>nvidia-container-cli: initialization error: nvml error: driver not loaded: unknown.<br/>