tensorflow

GPU tensorflow

install NVIDIA-CUDA

http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu-installation

$ wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb

$ sudo dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda

vim .bashrc

# cuda 9.1
export PATH=/usr/local/cuda-9.1/bin${PATH:+:${PATH}}

create symlink for 9.0

$ cd /usr/local/cuda-9.1/lib64
$ sudo ln -s libcublas.so.9.1 libcublas.so.9.0
$ sudo ln -s libcusolver.so.9.1 libcusolver.so.9.0
$ sudo ln -s libcudart.so.9.1 libcudart.so.9.0
$ sudo ln -s libcufft.so.9.1 libcufft.so.9.0
$ sudo ln -s libcurand.so.9.1 libcurand.so.9.0

install cuDNN

http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html

download@local

(local)$ scp libcudnn7_7.0.5.15-1+cuda9.1_amd64.deb jason@141.52.97.232:~/install/cuDNN_7/
(local)$ scp libcudnn7-dev_7.0.5.15-1+cuda9.1_amd64.deb jason@141.52.97.232:~/install/cuDNN_7/
(local)$ scp libcudnn7-doc_7.0.5.15-1+cuda9.1_amd64.deb jason@141.52.97.232:~/install/cuDNN_7/

$ sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.1_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.0.5.15-1+cuda9.1_amd64.deb
$ sudo dpkg -i libcudnn7-doc_7.0.5.15-1+cuda9.1_amd64.deb

Remove CUDA 9.1

$ sudo apt-get --purge remove cuda-9-1
$ sudo apt autoremove

Remove cuDNN for 9.1

Install CUDA9.0

https://developer.nvidia.com/cuda-90-download-archive

$ wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb

$ sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda-9-0

cuDNN 7

download 3 files @local

(local)$ scp libcudnn7* jason@141.52.97.232:~/install/cuDNN_7/

$ sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.0.5.15-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-doc_7.0.5.15-1+cuda9.0_amd64.deb