1. Check if CUDA capable

lspci | grep -i nvidia

2. Check if it has supported version of Linux

uname -m && cat /etc/*release

3. Check if gcc installed

gcc --version If missing, run

sudo apt install gcc -y

4. Install CUDA Toolkit

Head over to https://developer.nvidia.com/cuda-downloads and select your configurations

Select your installer type and follow the instructions

5. Install Nvidia Driver

Head over to https://www.nvidia.com/en-us/drivers/

Fill out your specifications

Take note of the Driver version

Install the driver