wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

Install libs for Caffe:

sudo apt-get update
 
sudo apt-get upgrade
 
sudo apt-get install -y build-essential cmake git pkg-config
 
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
 
sudo apt-get install -y libatlas-base-dev 
 
sudo apt-get install -y --no-install-recommends libboost-all-dev
 
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
 
# (Python general)
sudo apt-get install -y python-pip
 
# (Python 2.7 development files)
sudo apt-get install -y python-dev
sudo apt-get install -y python-numpy python-scipy
 
# (or,Python 3.5 development files)
sudo apt-get install -y python3-dev
sudo apt-get install -y python3-numpy python3-scipy