(TensorFlow) Windows 10 에 Tensorflow GPU 설치하기

Cuda 및 Tensorflow 프로그램을 실행하기 위해 Tensorflow-gpu 버전을 설치하는 과정을 소개합니다.

우선 가장 중요한건 모두 관리자 권한으로 실행하여 설치해야합니다.

또한 Python 3.5 버전이 제대로 동작합니다.

전 Windows 10 pro 에 Visual Studio 2017 과 Python 3.6 이 설치된 상태에서 진행했습니다.

 

1. Anaconda 4.2.0 설치

https://www.anaconda.com/download/

반드시 Python 3.5 기반으로 된 Anaconda 4.2.0 를 설치해야합니다.

위 링크를 가보면 아시겠지만 최신버전은 Python 3.6 을 기반으로 하여 Anaconda  5.0 버전까지 나온 상태입니다.

최신버전을 깔지 마세요!! (최신버전을 깔고 파이썬 3.5 를 설치하여 환경을 구성하는 방법도 있다고 하긴합니다... 만.. 안할래요..)

아래 링크로 들어가서

https://repo.continuum.io/archive/

Anaconda3-4.2.0-Windows-x86_64.exe 파일을 다운 받습니다.

다운 받은 파일을 관리자 권한으로 실행합니다.

아래는 설치 과정입니다. 특별한건 없습니다.

 

2. cuda_8.0.61_win10 설치

아래 페이지로 이동하여 cuda_8.0.61_win10 을 다운 받습니다.

자신의 운영체제에 맞게 선택한 후 다운로드 합니다.

(현재 최신버전이 바뀌어있는데..  cuda_8.0.61_win10 버전은 요쳥하시면 메일로 보내드리겠습니다.)

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

다운받은 파일을 관리자 권한으로 실행합니다.

요것도 특별한거 없이 설치하면 됩니다.

그래픽카드 드라이버 등을 새로 깔기때문에 모니터까 껐다 켜졌다 합니다.

(참고로 전 1050 Ti 를 사용하고 있습니다.)

 

[아래그림] visual studio 2015 가 설치되어있는 경우

또한 Cuda 프로그램은 설치 후 재설치를 하게되면 설치 중간에 먹통이 되더군요; (화면이 시꺼메지면서 먹통이 됩니다.;;;

다행히 재부팅하면 되긴하는데 이렇게 되면 먼가좀.. 꺼름직하므로 안된다고 재설치하는건 피하시길..

 

3. cudnn-8.0-windows10-x64-v5.1 다운 및 복붙하기

아래 링크로 들어가 5.1 버전을 다운 받습니다.

https://developer.nvidia.com/cudnn

위 그림에서 Download 를 선택하면

아래 그림과 같이 회원가입을 필요로 합니다.

가입 후 로그인합니다.

아래 그림처럼 멀 물어보는데 아무거나 선택하세요;;

Agree... 체크합니다.

체크하면 아래쪽으로 쭉 버전별 라이브러리가 나오는데 여기서 아래그림과 같이 5.1 버전을 다운 받습니다.

압축파일이 하나 받아지는데 이를 풀면 아래와 같은 파일이 보입니다.

bin, include, lib 폴더를 복사하고

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 위치에 붙여넣기합니다.

 

4. Tensorflow-GPU 설치

cmd 창을 관리자권한으로 실행합니다.

pip install tensorflow-gpu

위 명령어를 실행하면 아래 그림과 같이 설치파일 다운 및 설치가 진행됩니다.

마지막에 초록색으로 pip 버전을 upgrade 하라고 나오는데 해야합니다.

python -m pip install --upgrade pip

이제 python 을 실행하여 편집기로 아래 구문을 실행해 봅니다.

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

참고로 import tensorflow as tf 에서 에러가 발생된다면 설치가 제대로 안된것 입니다. ㅜㅠ

아래 그림을 보시면 아시겠지만

sess = tf.Session()

위 명령어를 실행할때 그래픽카드를 인지하는 걸 알 수 있습니다.

 

5. Jupyter 로 실행

Anaconda 에는 Jupyter Notebook 에디터가 있습니다.

이를 관리자 권한으로 실행합니다.

아래와 같은 그림이 나오는데 우측 상단의 New 를 선택해 Folder 를 하나만듭니다.

이름이 이상하게 되어있는데 Untitled Folder 를 체크하면 위쪽에 Rename 버튼이 활성화 됩니다.

버튼을 눌러 이름을 변경하고 폴더로 진입합니다.

진입한 후New 에서 Python [default] 를 선택합니다.

선택하게되면 코딩이 가능한 에디터 모양이 나옵니다.

이를 이용하여 아래 코드를 실행해 봅니다.

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

 

a = tf.constant(55)
b = tf.constant(37)
print(sess.run(a+b))

 

실행버튼은 Play 버튼을 누르면 결과가 나타납니다.

 

위와 같이 결과가 나오면 모든 설정이 완료된 것입니다.

 

참고

http://goodtogreate.tistory.com/entry/GPU-TensorFlow-on-Window-10-TensorFlow-GPU%EB%B2%84%EC%A0%84-%EC%9C%88%EB%8F%84%EC%9A%B010-%EC%84%A4%EC%B9%98

http://daeson.tistory.com/276

https://brunch.co.kr/@mapthecity/15

http://mataeoh.egloos.com/7052271

http://www.gurobi.com/downloads/get-anaconda

실행오류 관련

https://github.com/tensorflow/tensorflow/issues/9033

https://www.microsoft.com/ko-kr/download/confirmation.aspx?id=52685

 

* 파이썬 3.5 아나콘다를 사용하는 방법.

파이썬 3.5에서 아나콘다를 사용하는 방법에는 세 가지가 있습니다.
• 최신 버전의 Anaconda를 다운로드 한 다음 Python 3.5 환경을 만드는 것이 좋습니다.
• 또는 최신 버전의 Anaconda를 다운로드하고이 명령을 실행하여 루트 환경에 Python 3.5를 설치하십시오. conda install python = 3.5
• 또는 Python 3.5가 기본적으로 포함 된 최신 Anaconda 인 Anaconda 4.2.0을 다운로드하십시오. 우리의 아카이브에서 다운로드 할 수 있습니다. 플랫폼에 맞는 버전 4.2.0을 찾을 때까지 페이지를 아래로 스크롤하십시오.