- Newest
- Most votes
- Most comments
You can use Deep Learning AMI which comes pre-installed with NVIDIA drivers and other software, and are available at no additional charge. For example, search for Deep Learning Base AMI (Amazon Linux 2) when launching EC2 instance (screenshot below)
As AL2 will end support soon, you may want to consider a newer OS. There are DLAMIs for AL2023 and Ubuntu. If you need AL2023 or Ubuntu with GUI, try Deep Learning graphical desktop on Amazon Linux 2023 (AL2023) with AWS Deep Learning AMI (DLAMI) and Deep Learning graphical desktop on Ubuntu Linux with AWS Deep Learning AMI (DLAMI)
In addition to AMI with NVIDIA drivers, you can install NVIDIA driver yourself. Install NVIDIA GPU driver, CUDA Toolkit, NVIDIA Container Toolkit on Amazon EC2 instances running Amazon Linux 2 (AL2) includes an install script that you can copy and paste. Do ensure your AL2 instance have sufficient free disk space.
When looking for the Amazon Linux 2 AMI with NVIDIA Tesla GPU drivers, you have a few different options to find and use these AMIs.
If you're not seeing the specific NVIDIA Tesla GPU driver AMI in the EC2 console's "Browse more AMIs" section, but can see it in the general AWS Marketplace, this could be due to a few reasons:
-
AWS and NVIDIA offer different types of pre-configured AMIs with NVIDIA drivers installed, including Tesla driver offerings, GRID driver offerings, and Gaming driver offerings.
-
For AI/ML workloads, you might want to consider using the AWS Deep Learning AMIs, which come pre-configured with NVIDIA drivers and other tools needed for machine learning workloads. For example, the AWS Deep Learning Base GPU AMI (Amazon Linux 2023) includes the latest NVIDIA drivers, CUDA, Docker, and other necessary components for GPU-accelerated workloads.
-
If you're specifically looking for a Tesla driver AMI for compute workloads (like AI/ML), you can search directly in the AWS Marketplace rather than through the EC2 console's limited selection.
If you're experiencing issues with NVIDIA drivers on your instances, you can also install the drivers manually or use one of the Deep Learning AMIs that AWS provides, which typically come with the necessary drivers pre-installed and configured.
For AI/ML workloads specifically, the Deep Learning AMIs are often a better choice as they include not just the drivers but also frameworks and tools commonly used in machine learning.
Sources
NVIDIA drivers for your Amazon EC2 instance - Amazon Elastic Compute Cloud
AWS Deep Learning Base GPU AMI (Amazon Linux 2023) - AWS Deep Learning AMIs
