텐서플로우가 이제 윈도우에서 설치하여 사용할 수 있게 되었습니다.

윈도우에 텐서플로우 설치 방법은 아래의 순서로 진행하시면 됩니다.


1.설치환경

- 윈도우10 64bit

- Nvidia GTX 1080 또는 1070, 1060 등 CUDA를 지원하는 그래픽 카드

 

2. 설치 패키지 다운로드

 

2.1 아나콘다3 4.2 64bit 버전 다운로드

 https://www.continuum.io/downloads

아래의 다운로드 화면에서 Python 3.5 version, 64-BIT Installer를 다운로드 한다.


  

2.2. Nvidia CUDA 및 cuDNN SDK 다운로드

https://developer.nvidia.com/accelerated-computing-toolkit

아래의 다운로드 화면에서 CUDA 8.0, cuDNN을 다운로드 하시면 됩니다.
cuNDD은 개발자 회원가입을 해야 다운 받을 수 있습니다.

 

 

3. 패키지 설치하기

 

3.1 CUDA 설치하기

다운받은 cuda_8.0.44_win10.exe 파일을 클릭하여 PC에 설치한다.

설치가 완료되면 아래의 경로에 CUDA 패키지가 설치된다.

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0


 

 

3.2 cuDNN 설치하기

다운받은 cudnn-8.0-windows10-x64-v5.1.zip 파일을 압축해제 한다.

압축을 해제하면 bin, include, lib 폴더가 있다. 해당 폴더를 3.1에 설치한 CUDA 설치 경로에 복사하여 덮어쓰기를 한다.

 

3.3 아나콘다 설치하기

다운받은 Anaconda3-4.2.0-Windows-x86_64.exe 파일을 클릭하여 PC에 설치한다.

 

3.4 Tensorflow 설치하기

아나콘다 설치 메뉴에서 아나콘다 프롬프트를 관리자 모드로 실행한다.

아래의 명령어를 아나콘다 프롬프트에  "pip install tensorflow-gpu" 명령어를 입력한다. 설치 명령어를 입력하면 아래와 같이 Tensorflow 다운로드 및 설치가 진행된다.



업그레이드 명령어는 아래와 같다.

pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl


4. 설치 확인 및 실행

아나콘다 설치 메뉴 폴더에서 IPython을 선택하여 실행한다.

아래의 코드를 입력하면 Tensoflow가 정상적으로 동작하는지 간단히 확인할 수 있다.

 import tensorflow as tf
 tf.Session()


반응형

'IT > 머신러닝' 카테고리의 다른 글

머신러닝 입문을 위한 좋은 강의 소개  (0) 2016.05.10
AWS에 Tensorflow 설치하기  (0) 2016.05.10
블로그 이미지

조이풀 라이프

Lift is short, enjoy the life

,