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
댓글 없음:
댓글 쓰기