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