텐서플로우(tensorflow) pip, 소스코드 설치 정보

tensorflow의 1.2.1버전에 대해서 pip install 방법과

github로 다운로드해서 설치해서 빌드하는 방법에 대해서 간단히 요약 정리 하였습니다.

***** pip 설치 

## cuda 설치

## 다운로드 URL :  https://developer.nvidia.com/cuda-downloads

sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

## cuDNN 설치 및 복사

## (로그인필요)다운로드 URL : https://developer.nvidia.com/rdp/form/cudnn-download-survey

sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include

sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

## 패키지 설치

$ $ sudo apt-get install python-pip python-dev 

## GPU 버전의 tensorflow 설치

$ pip install tensorflow-gpu

## 바이너리 확인 URL  : https://www.tensorflow.org/install/install_linux#the_url_of_the_tensorflow_python_package

# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7, cuda 8.0 cuDNN 5.1

export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.2.1-cp27-none-linux_x86_64.whl

## tensorflow GPU버전 바이너리 업데이트

sudo pip install --upgrade $TF_BINARY_URL

## PATH 설정

export PATH=$PATH:/home/tensorflow/bin

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"

export CUDA_HOME=/usr/local/cuda

## 추가 설치

sudo apt-get install python-pip python-dev python-virtualenv

--- 밑에 3개는 안해도 될듯 

## Virtualenv 환경  활성화

virtualenv --system-site-packages ~/tensorflow

## bash 사용할 경우

source ~/tensorflow/bin/activate

## Virtualenv 환경  비활성화

deactivate

## 텐서플로우 홈페이지 pip 설치하기 위한 TF_BINARY_URL 정보

https://www.tensorflow.org/versions/r0.12/get_started/os_setup

# 설치 위치

/usr/local/lib/python2.7/dist-packages

## 설치한 파이썬 라이브러이 위치 확인

python -c 'import os; import inspect; import tensorflow; print(os.path.dirname(inspect.getfile(tensorflow)))'

## 테스트 (버전마다 다름)

$ python -m tensorflow.models.image.mnist.convolutional

## gpu가 설치되어있다면 상태확인해보면서 테스트 해볼것!

## tensorflow 버전 확인

$ pip freeze | grep tensorflow

tensorflow==1.2.1

tensorflow-tensorboard==1.3.1

$ python

>>> import tensorflow as tf

>>> tf.__version__

'1.2.1'

----------------------------------------------------------------

***** 소스코드 빌드 설치

$ git clone https://github.com/tensorflow/tensorflow

$ cd tensorflow

$ ./configure   ## cuda, cudnn 디폴트 설정이 아니면 설정 필요

$ cd ..

$ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

$ bazel-bin/tensorflow/tools/pip_package/build_pip_package /home/tensorflow/tensorflow_0718

$ sudo pip install tensorflow-1.2.1-cp27-cp27mu-linux_x86_64.whl   ## .whl 파일명은 다를수 있음

## 여기에 설치됨

/usr/local/lib/python2.7/dist-packages/tensorflow