Nvidia cuda and cuDNN setup in Ubuntu 16.04

setup.sh

# Update Ubuntu apt cache
sudo apt update

# Update all Ubuntu packages
sudo apt upgrade

# Install build-essentials
sudo apt install build-essential

# Download cuda SDK 8 from Nvidia website (.run file)
# https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run
cd ~/Downloads
wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run
chmod +x cuda_8.0.61_375.26_linux.run

# Now we need to blacklist nouveau before installing the driver.
echo blacklist nouveau | sudo tee -a /etc/modprobe.d/blacklist-nouveau.conf
echo blacklist lbm-nouveau | sudo tee -a /etc/modprobe.d/blacklist-nouveau.conf
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/blacklist-nouveau.conf
echo alias nouveau off | sudo tee -a /etc/modprobe.d/blacklist-nouveau.conf
echo alias lbm-nouveau off | sudo tee -a /etc/modprobe.d/blacklist-nouveau.conf
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

# Update kernel and initrd
sudo update-initramfs -u

# Reboot now
# At Login screen goto tty mode by pressing Ctrl+Alt+F1
# Enter credentials and login

# Now we need to stop lightdm
sudo service lightdm stop

# cd to Downloads folder and execute .run file
cd ~/Downloads
sudo bash cuda_8.0.61_375.26_linux.run --no-opengl-libs
# Remember --no-opengl-libs flag is very important, else it may result into login loop

# During the install:
# Accept EULA conditions
# Say YES to installing the NVIDIA driver
# SAY YES to installing CUDA Toolkit + Driver
# Say YES to installing CUDA Samples
# Say NO rebuilding any Xserver configurations with Nvidia.

# If in lsmod nvidia is not present
sudo modprobe nvidia

# Set Environment path variables in bashrc
echo 'export PATH=/usr/local/cuda-8.0/bin/:$PATH' | tee -a ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH' | tee -a ~/.bashrc
echo 'export CUDA_HOME=/usr/local/cuda-8.0' | tee -a ~/.bashrc
echo 'export CUDA_ROOT=/usr/local/cuda-8.0' | tee -a ~/.bashrc
echo 'export LIBRARY_PATH=/usr/local/cuda/lib64' | tee -a ~/.bashrc
source ~/.bashrc

# Check CUDA driver version:
nvcc -V

# At this point you can start lightdm and login to Ubuntu via gui
sudo service lightdm start

# Download cuDNN v5.1 from nvidia website
# https://developer.nvidia.com/cudnn
# Extract cudnn-8.0-linux-x64-v5.1.tgz you will get a folder "cuda"
cd ~/Downloads/
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

# At this point cuda and cuDNN installation is complete.
# Go ahead and test it using cuda samples or running any gpu based ML project