don't stop believing

Windows에 Tensorflow 설치하기 본문

Python/Tensorflow

Windows에 Tensorflow 설치하기

Tongchun 2017. 9. 30. 22:41

인공지능이란 것을 배워 봅시다.

그중에서 가장 유명한 tensorflow를 설치하고 기본 예제를 따라해 봅시다.


언제 Windows에 python 3.5 이상의 버전이 설치되어 있어야 합니다. 윈도우 cmd에 python 버전을 확인해 봅니다.

python --version
Python 3.6.2

자.. 여기서부터 문제가 있었습니다. tensorflow는 Windows버전의 경우 python 3.5.x에서만 실행이 되고 있습니다.

그러니 python 3.5.4 버전을 설치해 주고 환경변수의 path를 다시 잡아 줬습니다.


cmd 또는 PowerShell을 관리자 모드로 실행합니다.

그리고 다시 python 버전을 확인 합니다.

python --version
Python 3.5.4

3.5.4 버전이니 pip3는 기본으로 포함됩니다.

이제 pip을 업데이트 해줍니다. 저는 최신 버전이군요.

python -m pip install --upgrade pip
Requirement already up-to-date: pip in c:\users\jake\appdata\local\programs\python\python35\lib\site-packages

이제 pip3로 tensorflow를 설치해 봅시다.

pip3 install https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl


업그레이드도 해줍니다.

pip3 install --upgrade tensorflow

Could not find a version that satisfies the requirement tensorflow (from versions: )

No matching distribution found for tensorflow

만약 위와 같은 메시지가 나온다면 python 버전이 3.5.x 버전인지 확인해 봅니다. 제 경우 3.6.x 버전으로 tensorflow를 설치할 경우 발생했었습니다. 3.5.x에서는 발생하지 않습니다.

(3.5.2 버전에서도 발생하네요.)


정상적으로 설치가 되었다면 아래와같이 메시지가 나오는 것을 확인할 수 있습니다.

Successfully installed bleach-1.5.0 html5lib-0.9999999 markdown-2.6.9 numpy-1.13.3 protobuf-3.4.0 setuptools-36.5.0 six-1.11.0 tensorflow-1.3.0 tensorflow-tensorboard-0.1.7 werkzeug-0.12.2 wheel-0.30.0

tensorflow 말고도 설치되는 라이브러리들이 많습니다.


만약 nvidia GPU를 사용하고 있다면 엔비디아 사이트에서 CUDA 툴킷을 설치한 뒤 다음 명령어로 쉽게 GPU 가속을 지원하는 tensorflow를 설치할 수 있습니다.

pip3 install --upgrade tensorflow-gpu

tensorflow에서 자주 사용하는 라이브러리들도 설치를 해줍니다.

pip3 install numpy matplotlib pillow

numpy는 수치 계산 라이브러리, matplotlib는 그래프 출력 라이브러리, pillow는 이미지 처리 라이브러리 입니다.

이제 tensorflow가 재대로 설치되었는지 확인해 봅시다.

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

위 코드를 실행하면 b'Hello, TensorFlow!'가 출력되는 것을 확인 할 수 있습니다.


여기까지 tensorflow 설치였습니다.


'Python > Tensorflow' 카테고리의 다른 글

openAI gym 설치하고 sample 따라하기 (on Mac)  (0) 2017.11.28
인공지능 학습 자료  (0) 2017.11.21
Mac에 TensorFlow 설치  (0) 2017.10.18
Comments