How to install Nvidia CUDA driver on AWS ec2 instance?

I am using AWS instance but Nvidia CUDA driver is not install there. I need help to install Nvidia CUDA driver on ec2 install.

The Ubuntu version is 18.04 and the instance is GPU based. The graphics card is nvidia t4. I am using the g46nxlarge instance.

1 Answer

The g4dn.xlarge EC2 instances are not your typical virtual machines, as there are some unique features that enable the machine to interface directly with a dedicated video card. For this reason, there are dedicated AMIs for Deep Learning, virtual workstations, and gaming. However, if you insist on starting from scratch, Amazon has provided a step-by-step that shows you how to get the CUDA drivers installed on your instance.

The gist goes like this:

1. Visit the NVIDIA Driver Search page to search for the appropriate driver package for your installation:
2. Install the driver using the steps from NVIDIA's Data Center Documentation, not forgetting the myriad post-installation steps

Given the number of steps and ways the manual installation can go wrong, you might want to start with the pre-built AMIs first and then rebuild with a custom installation (if required) later.