Installing GPU libraries and drivers

Now, let's get our computers set up for building AI applications. If you wish to do this task on your local computer, it's advised that you have either Windows or a Linux distribution installed. Unfortunately, most macOS are not built to accommodate GPUs, and therefore we will not be touching on macOS in this section. If you do not have an NVIDIA GPU on your computer, please perform the following steps. If you do have an NVIDIA GPU, you may choose to follow along with the AWS-based section, or skip to the following section.

If you're not sure whether you have an NVIDIA GPU, you can check for its existence with the following terminal command: 

lspci -nnk | grep -i nvidia