Notice
Recent Posts
Recent Comments
Today
Total
05-04 00:14
Archives
관리 메뉴

Jeongchul Kim

WINDOWS 10 TensorFlow GPU 본문

Windows 10

WINDOWS 10 TensorFlow GPU

김 정출 2017. 1. 6. 11:57


WINDOWS 10 TensorFlow GPU



Tensorflow v0.12.1 GPU

Anaconda 4.2.0

CUDA 8.044

cuDNN 5.1


Tensorflow Install Doc

https://www.tensorflow.org/get_started/os_setup#pip-installation-on-windows


Anaconda 4.2.0 Install

Numpy, SciPy, Matplotlib 등 연산에 필요한 Python Package 들을 통합했다.


https://www.continuum.io/downloads


Python 3.5 version 64-BIT INSTALLER 를 다운 받습니다.



Next 버튼 -> I Agree 버튼-> Next버튼 -> Next버튼 -> Install 버튼을 클릭합니다.



Finish 버튼을 클릭합니다.



CUDA 8.044 -win 10 설치

TensorFlow v0. 12. 0 RC1는 CUDA 8.0 이상 설치를 권장한다.

local 버전 1.2 GB 직접 다운받아 설치한다.

https://developer.nvidia.com/cuda-downloads



다운 받은 CUDA를 Setup 합니다.

cuDNN 5.1 Install

Deep Learning을 위해 NVIDIA에서 제공하는 라이브러리다.

cuDNN 5.1 을 설치한다.

https://developer.nvidia.com/cudnn


다운로드를 받기 위해서 NVIDIA에 회원가입을 해야 한다.



회원가입 후에 로그인을 한다.



Download 버튼을 클릭합니다.



cuDNN Download Survey 를 입력하고 Proceed To Downloads 버튼을 클릭합니다.



I Agree To the Terms of the cuDNN Software License Agreement 체크박스를 선택하면

다운 가능한 목록에서 cuDNN v5.1 for CUDA 8.0 을 클릭합니다.



OS 환경 Window 10을 클릭합니다.



다운받은 압축 파일을 압축 해제 합니다.


압축 해제 후에 bin, include, lib 안에 있는 파일을 CUDA 설치 경로(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)에 각각 붙여 넣습니다. 관리자 권한을 요구합니다.

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

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

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


TensorFlow-GPU Install

cmd 명령프롬프트를 실행한다.

다음의 명령을 입력한다.


C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl



설치 도중에

Cannot remove entries from nonexistent file c:\users\user\anaconda3

관련 오류가 나온다면 다음의 명령어를 실행하시기 바랍니다.


C:\> pip install --ignore-installed --upgrade  https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl



Tensorflow 실행

cmd 명령 프롬프트에서 python 으로 들어가

import tensorflow as tf 를 입력합니다.

다음의 명령어를 입력합니다.


hello = tf.constant(‘Hello, TensorFlow GPU’)

sess = tf.Session()

print(sess.run(hello))

네 이상 마치겠습니다.




Comments