Tensorflow GPU版インストールメモ（Ubuntu16.04 + GeForce GT 730M）

環境

• Ubuntu 16.04
• Thinkpad t540p
• NVIDIA GeForce GT 730M

必要なソフトウェアのインストール

sudo apt-get install openssh-server git vim libssl-dev

nouveau を無効化

sudo sh -c "cat << ETX > /etc/modprobe.d/blacklist-nouveau.conf
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
ETX" && cat /etc/modprobe.d/blacklist-nouveau.conf

sudo sh -c "echo 'options nouveau modeset=0' > /etc/modprobe.d/nouveau-kms.conf" && cat /etc/modprobe.d/nouveau-kms.conf
sudo update-initramfs -u
sudo reboot

cudaのインストール

sudo service lightdm stop
wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_linux-run
chmod 755 cuda_8.0.44_linux-run
sudo ./cuda_8.0.44_linux-run --override
sudo apt-get install nvidia-367 --reinstall
sudo reboot

cuDNNのインストール

https://developer.nvidia.com/rdp/cudnn-download
登録後、ダウンロード

tar zxvf cudnn-8.0-linux-x64-v5.1.tgz cuda/
sudo cp -a cuda/lib64/* /usr/local/lib/
sudo cp -a cuda/include/* /usr/local/include/
sudo ldconfig

pyenvのインストール

git clone https://github.com/yyuu/pyenv.git ~/.pyenv

設定を追加

echo 'export PYENV_ROOT=$HOME/.pyenv' >> ~/.bash_profile
echo 'export PATH=$PYENV_ROOT/bin:$PATH' >> ~/.bash_profile
echo 'eval "$(pyenv init -)"' >> ~/.bash_profile
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bash_profile
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bash_profile

設定を適用

source .bash_profile

Python&Tensorflowのインストール

pyenv install 2.7.11
pyenv global 2.7.11
pip install tensorflow-gpu

pyenv install 3.5.1
pyenv global 3.5.1
pip install tensorflow-gpu