Remove Default Driver
sudo apt –purge remove xserver-xorg-video-nouveau
Add NVIDIA Respository
sudo add-apt-repository ppa:graphics-drivers/ppa
Install NVIDIA Driver
sudo apt install nvidia-361 nvidia-settings nvidia-prime
Install NVIDIA CUDA
sudo apt install nvidia-cuda-toolkit