tensorflow 설정

근 1년 만에 다시 시작하려니 개발환경 셋팅하는 것도 낯설다.
또 잊기 전에 정리해두자.

현재 개발 환경은
ubuntu 18.04 LTS
nvidia GTX 1080 Ti

1. gpu 드라이버 설정

아래 둘 중에 한가지 방법으로 설정.
1)
$ ubuntu-drivers devices 내 기기 확인
$ sudo apt install nvidia-driver-390 설치
$ nvidia-smi 로 확인

2)
$ add-apt-repository ppa:graphics-drivers/ppa
$ apt update
$ apt install nvidia-390
$ nvidia-smi

2. cuda 9.0 

nvidia 사이트에  가서 .deb 파일을 다운받고 설치.

$ dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
$ apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
$ apt-get install cuda

3. .bashrc 수정

export PATH=$PATH:/usr/local/cuda/bin
export CUDADIR=/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

4. cudnn 7.0
( https://developer.nvidia.com/rdp/cudnn-archive )

$ tar xvfz cudnn-9.0-linux-x64-v7.tgz
$ cp cuda/include/* /usr/local/cuda/include
$ cp -d cuda/lib/* /usr/local/cuda/lib64
$ chmod a+r /usr/local/cuda/include/cudnn.h
$ chmod a+r /usr/local/cuda/lib64/libcudnn*

patch 파일들을 다 받아서 순서대로

dpkg -i 패키지.deb 로  설치.

5. tesorflow-gpu 1.10.1

텐서플로우 설치.

$ pip3 install tensorflow-gpu