Tensorflow 설치 ( Ubuntu 18.04)

사실 대 AI 시대를 맞이하여 tensorlfow 설치를 하고 얘기들이 많지만

생각보다 tensorflow 관련해서 배우기보다는 채굴 커뮤니티에서 훨씬 많이 배운거 같다.

 

overflow 제한 푸는 방법과 설치 법 등 훨씬 다양한 정보가 많다.

생각보다 재야의 고수가 많아 좋은 정보와 글들이 넘쳐난다.

 

본론으로 넘어가서 Nvidia 비디오 카드인 geforce dirve를 사용하기 위하여 cuda 설치 및 설정이 필요하다.

하다가 보니 굉장히 많은 삽질들과 안되는 경우가 다양해서 성공한 경우를 일단 기록한다.

 

Ubuntu 18.04 TensorFlow setting

 

OS : 18.04

GPU : Geforce 1060 6GB

 

1. Nvidia driver install

• 기존 드라이버 설치 확인

• cat /proc/driver/nvidia/version

• 기존 그래픽 드라이버 삭제

• sudo apt-get autoremove nvidia-* && sudo apt-get purge nvidia-*

• 그래픽카드 모델에 맞는 버전 확인

• 링크를 통해 버전 확인

• https://www.nvidia.com/Download/Find.aspx?lang=en-us

• 18.04는 레포짓터리 추가 필요

• sudo add-apt-repository ppa:graphics-drivers/ppa

• sudo apt-update

• sudo apt-get install nvidia-390

1. Cuda 9.0 설치

• sudo apt install -y gcc-6 g++-6

• 링크 통해 다운로드

• https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=debnetwork

 

• Local로 run 파일 다운로드 시 

• sudo chmod +x cuda_9.0.176_384.81_linux.run

• ./cuda_9.0.176_384.81_linux.run --override

Do the CUDA post-install actions

So Tensorflow can find your CUDA installation and use it properly, you need to add these lines to the end of you ~/.bashrc or ~/.zshrc.

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

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

 

• sudo apt-get install libcudnn7-dev

• python3 import tensorflow 에러

• cudnn 미설치 에러

 

python3: Relink `/lib/x86_64-linux-gnu/libudev.so.1' with `/lib/x86_64-linux-gnu/librt.so.1' for IFUNC symbol `clock_gettime' Segmentation fault (core dumped)

Step 4: Install libcupti

This one is easy.

sudo apt-get install libcupti-dev

1. python 설치

• python2 또는 python3 설치

• python2는 업데이트 중지 되어 python3를 설치함

• sudo apt-get install python3 python3-pip

• TensorFlow 설치

• CPU 버전

• pip3 install tensorflow

• GPU 버전

• pip3 install tensorflow-gp