Google Cloud Platform の GPU インスタンスの Ubuntu 16.04 LTS に TensorFlow をインストール

資料

自分用メモ。使っている GPU は NVIDIA Tesla K80 を 1 枚。

# CUDAやドライバ諸々入れてもらう
sudo curl -O 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 -y
sudo reboot

# 動作確認
nvidia-smi

# cuDNNを落としてくる
# CUDA 8.0ならcuDNN 5.1以上らしい
gsutil cp gs://.../cudnn-8.0-linux-x64-v5.1.tgz .
tar xvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp -a cuda/lib64/* /usr/local/cuda-8.0/lib64/
sudo cp -a cuda/include/* /usr/local/cuda-8.0/include/
sudo ldconfig

# pipとかをインストール
sudo apt-get install python3-pip python3-dev

# localeがja_JP.UTF-8だとpipがちゃんと動かないのでen_US.UTF-8にする（クソ）
export LC_ALL="en_US.UTF-8"
export LC_CTYPE="en_US.UTF-8"
sudo dpkg-reconfigure locales

# GPU用のTensorFlowをインストール
sudo pip3 install tensorflow-gpu