2017년 2월 27일 월요일

텐서플로우 설치 (TensorFlow for windows)

1. 파일 다운로드

  - Anaconda 4.3.0 for windows : https://www.continuum.io/downloads

  - CUDA 8.0.44 : https://developer.nvidia.com/cuda-downloads
      local 버전 권장 ( network 버전은 비교적 오래걸림 )

  - NVIDIA cuDNN : https://developer.nvidia.com/cudnn
      deep learning 용 추가 라이브러리 ( CUDA8.0 은 cuDNN 5.1 설치 )
      회원가입 필요

  - Python 3.5 : https://www.python.org/downloads/release/python-350/
      윈도우에서 tensorFlow 를 시작하려면 3.5 버전이상 필요


2. Anaconda 4.3.0 , CUDA 8.0, Python3.5 설치
    python 설치 시 path 등록에 체크하면 환경변수 등록하는 작업을 생략할 수 있다.


3. cuDNN 압축 해제 후 CUDA 8.0 의 경로에 붙여넣기
    CUDA 경로 : C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0


4. TensorFlow-GPU 설치
 
  1) cmd 관리자 실행하여 다음 명령어를 입력한다.
   pip install tensorflow-gpu

  2) 이전의 명령어에서 pip 버전이 낮아서 불가하다는 문구가 나오면 다음명령어로 pip를 업그레이드 한 후에 다시 1) 을 수행한다.
   python -m pip install --upgrade pip

  3) tensorflow 가 성공적으로 설치가 완료되면 다음의 명령어를 수행한다.
   pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl


5. 설치 확인

   Anaconda prompt를 관리자 권한으로 실행하고 Python 에서 다음의 code 를 입력한다.

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

   결과 : Hello, TensorFlow


텐서플로우 설치에 참고할만한 사이트
1. 텐서플로우 문서 한글 번역본
https://tensorflowkorea.gitbooks.io/tensorflow-kr/content/g3doc/get_started/os_setup.html
2. 블로그
http://comajava.blogspot.kr/2017/01/windows-tensorflow-python-35-anaconda.html

댓글 없음:

댓글 쓰기