[TensorFlow] How to install tensorflow-gpu on ubuntu16.04

1) Nvidia 드라이버 설치

$ sudo add-apt-repository ppa:graphics-drivers/ppa 
$ sudo apt-get update 
$ sudo apt-get install nvidia-375
$ sudo reboot

이후 다음과 같이 확인 할 수 있다.

2) CUDA Toolkit 8.0 설치

https://developer.nvidia.com/cuda-downloads

$ sudo sh cuda_8.0.61_375.26_linux.run

설치 과정 중 질문에 대해서는 다음과 같이 답하면 된다.

추가적으로 다음 글을 참고하여 환경구성을 통해 라이브러리와 CUDA를 사용 할 수 있도록 경로를 추가하자.

2017/07/26 - [리눅스/ubuntu] - Cuda 8.0 toolkit install - nvcc not found - ubuntu 16.04

3) CuDNN v5.1 설치

https://developer.nvidia.com/cudnn

Download 

cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0

cuDNN v5.1 Library for Linux

$ tar xzvf cudnn-8.0-linux-x64-v5.1.tgz
$ sudo cp cuda/lib64/* /usr/local/cuda-8.0/lib64/ 
$ sudo cp cuda/include/* /usr/local/cuda-8.0/include/ 
$ sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn* 
$ sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h
$ sudo apt-get install libcupti-dev

4) Bazel 설치

2017/08/14 - [OS/Ubuntu] - (Ubuntu16.04) Bazel 설치

5) tensorflow-gpu 설치

(tensorflow) $ pip install --ignore-installed --upgrade \
https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.2.1-cp35-cp35m-linux_x86_64.whl

※ CPU

(tensorflow) $ pip install --ignore-installed --upgrade \
https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.2.1-cp35-cp35m-linux_x86_64.whl