Ubuntu16.04 install CUDA 9.0 + cuDNN 7.0.5

Step 0 Noveau drivers

1. Press ctrl+alt+F1
2. Create a file: /etc/modprobe.d/blacklist-nouveau.conf

sudo vi /etc/modprobe.d/blacklist-nouveau.conf

3. Put the following in the file

blacklist nouveau
options nouveau modeset=0

4. update-initramfs

sudo update-initramfs –u

5. reboot

sudo reboot

6. After reboot

lsmod | grep nouveau

And there is nothing output. If you output something, you need to check the step.

Step 1 Installation CUDA

1. update apt-get

sudo apt-get update

2. Install apt-get deps

sudo apt-get install openjdk-8-jdk git python-dev python3-dev python-numpy python3-numpy build-essential python-pip python3-pip python-virtualenv swig python-wheel libcurl3-dev curl

3. Install nvidia drivers

# The 16.04 installer works with 16.10.
# download drivers
curl -O http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb

# download key to allow installation
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

# install actual package
sudo dpkg -i ./cuda-repo-ubuntu1604_9.0.176-1_amd64.deb

#  install cuda (but it'll prompt to install other deps, so we try to install twice with a dep update in between
sudo apt-get update
sudo apt-get install cuda-9-0

4. reboot

sudo reboot

5. Check nvidia driver install

nvidia-smi

Step3 Install cuDNN

1. Install cuDNN

wget https://s3.amazonaws.com/open-source-william-falcon/cudnn-9.0-linux-x64-v7.1.tgz  
sudo tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz  
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

2. Open the ~/bashrc

sudo vi ~/.bashrc

3. Put the following in the end of the file:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:/usr/local/cuda/bin

4. Reload bashrc

source ~/.bashrc

Reference

[1] https://github.com/williamFalcon/tensorflow-gpu-install-ubuntu-16.04