動作確認済み

Ubuntu 14.04 + GeForce GTX TITAN X
Ubuntu 14.04 + GeForce GTX 1070

事前準備

環境はまっさらな方がいい.
cudaを入れるとnvidia driverも一緒に入るので、
nvidia driverは消しておいたほうがよい。

sudo apt-get remove --purge nvidia-375 nvidia-375-dev nvidia-modprobe nvidia-opencl-icd-375 nvidia-prime nvidia-settings -y

cudaをダウンロード&インストール

CUDA Toolkit 8.0 GA2　を選択（9.0は、要16.04）.
Linux -> x86_64 -> Ubuntu -> 14.04 -> deb (local)
と進み、
Base installerをダウンロードして、instructionに書いてある通り以下を実行.

sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

以下を.bashrcに記述.cudaのバージョンは自分の環境を確認し適宜変更.

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

cudaの確認

$ nvcc --version

を実行して,

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61

みたいに表示されたらok

cudnnをダウンロード&インストール.

https://developer.nvidia.com/cudnn
7.0以上はまだ対応していないもの(chainerなど)も多いため、6.0が推奨(2017/10/12)

Download -> アカウント作成 -> 同意 ->
-> Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0
-> cuDNN v6.0 Library for Linux

解凍して、cudnn関係のファイルを, /usr/local/以下にcpしてパスを通す.

cd Downloads
tar zxvf cudnn-8.0-linux-x64-v6.0.tgz 
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/* /usr/local/cuda/lib64/

cudnnの確認

/usr/local/cuda/include/cudnn.h

の

# define CUDNN_MAJOR      6
# define CUDNN_MINOR      0
# define CUDNN_PATCHLEVEL 21

で、versionが確認できる.