[Tensorflow] 우분투(Ubutu) 16.04에 텐서플로(Tensorflow) 설치하기

텐서플로를 설치하는 글을 많지만 정리하는 느낌으로 포스팅을 합니다.

 

텐서플로의 설치 안내 페이지에 면 버전이나 환경에 따라 설치하는 방법이 나와 있습니다.

(참고 : tensorflow.org - Download and Setup )

 

제가 설치한 환경입니다.

Ubuntu 16.04 64bit Desktop

GPU 없음

 

빠르게 테스트 환경을 구축하는 것을 목표로 했기 때문에 최대한 적게 설치하고 일반적으로 사용하는 버전 위주로 설치하겠습니다.

 

1. PIP 설치

 

텐서플로를 설치하기 위해서는 PIP를 설치해야 합니다.

(참고 : tensorflow.org - Pip Installation )

다음 명령어를 터미널에서 입력합니다.

sudo apt-get install python-pip python-dev

 

 

 

2. 텐서플로 설치

터미널에 아래 명령어를 입력하여 텐서플로 파일을 받을 주소를 세팅해 줍니다.

우리는 'CPU 전용', '파이썬 2.7'을 설치하겠습니다.

 

# Ubuntu/Linux 64-bit, CPU 전용, Python 2.7 - 우리는 이것만 사용합니다.
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl

# Ubuntu/Linux 64-bit, GPU 활성, Python 2.7
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl

# Ubuntu/Linux 64-bit, CPU 전용, Python 3.4
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.10.0-cp34-cp34m-linux_x86_64.whl

# Ubuntu/Linux 64-bit, GPU 활성, Python 3.4
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp34-cp34m-linux_x86_64.whl

# Ubuntu/Linux 64-bit, CPU 전용, Python 3.5
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.10.0-cp35-cp35m-linux_x86_64.whl

# Ubuntu/Linux 64-bit, GPU 활성, Python 3.5
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp35-cp35m-linux_x86_64.whl

 

이제 아래 명령어를 터미널에 입력하여 텐서플로를 설치합니다.

우리는 '파이썬 2'를 기준으로 설치합니다.

# Python 2 - 우리는 이것만 사용합니다.
$ sudo pip install --upgrade $TF_BINARY_URL

# Python 3
$ sudo pip3 install --upgrade $TF_BINARY_URL

 

 

아래와 같이 설치가 진행됩니다.

 

 

 

3. 테스트 코드 사용해보기

터미널 창에 입력해서 테스트해봅시다.

(참고 : tensorflow.org - Test the TensorFlow installation )

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

결과 : Hello, TensorFlow!

 

 

 

>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>

결과 : 42

 

 

이렇게 결과가 나온다면 텐서플로가 wp대로 설치된 것입니다.

 

마무리

이대로 텐서플로를 사용하기는 힘듭니다.

다음 포스팅에서는 파이참(PyCharm)이나 이클립스(Eclipse)를 설치하여 텐서플로를 사용해 보겠습니다.