2018년 1월 14일 일요일

TensorFlow 설치

참고 사이트
https://www.tensorflow.org/install/install_linux
https://tensorflowkorea.gitbooks.io/tensorflow-kr/content/g3doc/get_started/os_setup.html

AWS 프리티어 이용해서 우분투 EC2 인스턴스에서 테스트 중

Python : 2.7
TensorFlow : 1.4.1

python이랑 python pip 설치
sudo apt-get install python-pip python-dev
pip 이용해서 tensorflow 설치
자신의 환경에 맞는 tensorflow 버전을 선택해야 한다.

https://www.tensorflow.org/install/install_linux#the_url_of_the_tensorflow_python_package

여기에서 python 버전과 CPU만 사용할 것인지? GPU까지 사용할 것인지 선택해서 주소를 복사한다.
나는 python 2.7에 CPU만 사용할 거니깐.
https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.4.1-cp27-none-linux_x86_64.whlㅅ

현재(2018-01-14) 기준으로 1.4.1 버전이 최신 버전? 또는 stable 버전인 듯
문서에서는 변수 만들어서 URL 할당하고 하라는데 난 걍 ㄱㄱ

sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.4.1-cp27-none-linux_x86_64.whl

이렇게 하면 우선 설치는 끝?

정상적으로 설치되었고, 설치한 버전이 맞는지 확인하기 위해서
python을 치고 python 명령창으로 들어간다.

import tensorflow as tf
tf.__version__

위의 명령어를 쳤을 때 아까 설치한 버전이 나오면 정상적으로 설치 완료.


간단한 예제 프로그램 돌리기

Hello, TensorFlow!를 찍어 주는 프로그램

위에서 tensorflow를 import했다고 가정하고,
hello = tf.constant("Hello, TensorFlow!");
sess = tf.Session();
print(sess.run(hello))

이렇게 하면 Hello, TensorFlow!라 화면에 찍힌다.

이때 tf.Session()에서 에러가 나는 경우가 있는데(아마도 CPU 이용해서 하는 경우 다 나는 것 같다.), 이건 현재 CPU에 최적화되어서 compile되지 않아서 성능이 제대로 안나올 수 있다는 경고 정도이다. (I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA)
이걸 없애고 싶으면, 소스 받아서 직접 컴파일해서 써야 되는데 귀찮...

댓글 없음:

댓글 쓰기