AI / Google TensorFlow Install

2015년 인공지능 개발용 오픈소스 TensorFlow 를 공개했다.

TensorFlow

TensorFlow는 머신러닝을 위한 Research cloud로 1000 Cloud TPU로 가동하고 있다. 1000 TPU는 180Petaflops를 제공하고 있다.

Cloud TPU

Ubuntu Linux, macOS X, Windows 머신에 설치할 수 있다. Ubuntu 14.04 이상에서 설치 가능하다.

Install

먼저 TensorFlow 종류를 결정한다.


NVIDIA GPU에서 TensorFlow 실행시 요구조건

GPU 지원 TensorFlow를 설치하려면 아래 NVIDIA 소프트웨어가 설치되야 한다:

$ sudo apt-get install libcupti-dev

시스템에 설치된 소프트웨어가 위에 명시하 패키지 보다 이전 버전이면 업그레이드가 필요하다. 아닌 경우 TensorFlow가 실행되지만 아래 작업을 해주어야 한다:


TensorFlow 설치 방법

TensorFlow는 virtualenv, “native” pip, Docker, Anaconda 에서 설치할 수 있다. virtualenv 환경을 권장한다.

virtualenv 환경에서 설치

여기서는

를 참조해서 virtualenv, virtualenvwrapper 를 구성해서 사용한다고 가정한다.

$ mkvirtualenv --system-site-packages -p python3 tensorflow
(tensorflow)$ 

그리고 tensorflow 패키지를 설치한다 - GPU 버전을 선택적으로 설치한다.

(tensorflow)$ pip install --upgrade tensorflow      # CPU
(tensorflow)$ pip install --upgrade tensorflow-gpu  # GPU

위 명령으로 설치가 실패시 pip 버전이 8.1 이하일 수 있다. 업그레이드 후 재 시도한다. 혹은 다음 명령으로 TensorFlow Python package를 직접 설치한다.

(tensorflow)$ pip install --upgrade tfBinaryURL

tfBinaryURL여기서 찾아서 직접 지정한다.

Uninstall Tensorflow

가상환경 디렉토리를 삭제하거나

(tensorflow)$ deactivate
$ rm -rf .virtualenv/tensorflow

혹은 가상환경에서 tensorflow 를 pip로 지운다.

(tensorflow)$ pip uninstall tensorflow      # CPU
(tensorflow)$ pip uninstall tensorflow-gpu  # GPU


Run a short TensorFlow program

python 을 실행해 REPL 환경 혹은 스크립트로 아래 코드를 입력한다:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

결과로 b'Hello, TensorFlow!' 같이 출력되면 성공적으로 실행된 것이다.

이어서 Getting Started with TensorFlow 를 따라간다.

참조