Ubuntu에 GPU 지원 TensorFlow 설치하기 - 1

리눅스 머신에 TensorFlow를 설치하기 위해서는 다음과 같은 기본 조건이 필요

• 64-bit desktops or laptops

• Ubuntu 16.04 or higher

GPU 지원 TensorFlow를 실행시키기 위한 NVIDIA 요구사항

• CUDA Toolkit 9.0. For details, NVIDIA 설명서 참조. NVIDIA 설명서에 설명 된대로 관련 CUDA 경로 이름을 LD_LIBRARY_PATH 환경 변수에 추가해야합니다.
• CUDA Toolkit 설치전에 NVIDIA graphic driver를 설치- 설치방법
• CUDA Toolkit 9.1 download

$ wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda  

• 설치후 환경설정
• /usr/local/cuda-9.1/bin 을 path에 추가

$ export PATH=/usr/local/cuda-9.1/bin${PATH:+:${PATH}}

• /usr/local/cuda-9.1/lib64 을 LD_LIBRARY_PATH에 추가

$ export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

• cuDNN SDK v7. For details, see NVIDIA's documentation. CUDA_HOME 환경 변수를 만들 었는지 확인 필요.
• developer.nvidia.com에서 사용자 등록
• NVIDIA cuDNN home page 접속 “DOWNLOAD cuDNN” Click
• Software License Agreement 체크 후 Download cuDNN v7.1.3 (April 17, 2018), for CUDA 9.1 선택
•  cuDNN v7.1.3 Runtime Library for Ubuntu16.04 (Deb) , cuDNN v7.1.3 Developer Library for Ubuntu16.04 (Deb) 다운로드
• 런타임 라이브러리 인스톨

$ sudo dpkg -i libcudnn7_7.1.3.16-1+cuda9.1_amd64.deb 

• 개발자 라이브러리 인스톨

$ sudo dpkg -i libcudnn7-dev_7.1.3.16-1+cuda9.1_amd64.deb 

•    샘플 코드와 유저가이드 설치

$ sudo dpkg -i libcudnn7-doc_7.1.3.16-1+cuda9.1_amd64.deb 

•  
• GPU drivers supporting your version of the CUDA Toolkit.
• GPU card with CUDA Compute Capability 3.0 or higher for building from source and 3.5 or higher for our binaries. See NVIDIA documentation for a list of supported GPU cards.
• GPU card with CUDA Compute Capability 3.0 or higher
• The libcupti-dev library,
CUDA Toolkit >= 9.0:

$ sudo apt-get install cuda-command-line-tools 

and add its path to your LD_LIBRARY_PATH environment variable:


$ export LD_LIBRARY_PATH=${LD_LIBRARY_PATH:+${LD_LIBRARY_PATH}:}/usr/local/cuda/extras/CUPTI/lib64

For CUDA Toolkit <= 7.5 do:


$ sudo apt-get install libcupti-dev 

Docker 에 TensorFlow 설치하기

Reference

•  https://www.tensorflow.org/install/install_linux#NVIDIARequirements