본문 바로가기

Tech/머신러닝

[윈도] 딥러닝 개발환경 만들기 - TensorFlow 설치

네이버 공유하기
728x90

[윈도] 딥러닝 개발환경 만들기 - TensorFlow 설치


1. 디스플레이 드라이버 설치 

보통 윈도에서 제공하는 드라이버를 사용하기도 하는데, 혹시 모르니 공식 드라이버를 설치한다.


2. Cuda Toolkit 설치

TensorFlow 를 보면 CPU 만 이용하는 방법이 있고, GPU 도 같이 이용하는 방법이 있다.

GPU 도 같이 사용하려면 Cuda 가 필요한데, 그렇다면 Nvidia 그래픽 카드에서만 GPU 사용이 가능한가보다.

현재는 쿠다 8.0 만 지원한다.

https://developer.nvidia.com/cuda-80-ga2-download-archive


3. Python 설치

아직도 2.x 과 3.x 에서 고민중인데, 텐서플로우에서는 3.x 을 사용한다. 

그러면 이제 3.5 냐 3.6 이냐가 고민이다;; (그냥 최신버전 3.6 설치)

https://www.python.org/downloads/

윈도 64비트는 https://www.python.org/downloads/windows/ 여기에서 볼 수 있고, (64비트 중요)

현재 최신 버전은 https://www.python.org/ftp/python/3.6.4/python-3.6.4-amd64.exe <-- 이걸 다운받으면 된다. 



4. TensorFlow 설치

C:\> pip3 install --upgrade tensorflow-gpu <-- GPU 를 이용하는 경우 

C:\> pip3 install --upgrade tensorflow <-- GPU 를 이용하지 않는 경우 (GPU 를 이용하는것보다 많이 느리다)


참고 : tensorflow 1.5 가 릴리즈되었습니다.
쿠다 9 을 지원한다고 하는데, 아직 설치해보지는 못했습니다만 뭐 그냥 설치하면 되지 않을까요...
하지만, 텐서플로우 홈페이지보니 https://www.tensorflow.org/versions/  1.4 버전이 안정버전이라고 하네요.

5. 설치 확인

$ python

Enter the following short program inside the python interactive shell: (에러가 없어야 한다)


>>> import tensorflow as tf

>>> hello = tf.constant('Hello, TensorFlow!')

>>> sess = tf.Session()

>>> print(sess.run(hello))

>>> print(tf.__version__)


참고 : https://www.tensorflow.org/install/install_windows


6. IDE 

파이참 : https://www.jetbrains.com/pycharm/download/#section=windows

주피터 : http://jupyter.org/

참고 : http://dwfox.tistory.com/69


반응형
네이버 공유하기


* 쿠팡 파트너스 활동을 통해 일정액의 수수료를 제공받을 수 있습니다.