To add the drivers repository to Ubuntu, run the commands below:

sudo add-apt-repository ppa:graphics-drivers/ppa

Update and Install Nvidia Drivers

sudo apt update

sudo apt install nvidia-396

• you can use Tab to see more updated drivers

For CUDA and related installations, see: https://medium.com/@zhanwenchen/install-cuda-and-cudnn-for-tensorflow-gpu-on-ubuntu-79306e4ac04e