Windows에 Tensorflow 설치하기

1. 텐서플로우(Tensorflow)란?

– 기계학습과 딥러닝을 위해 구글에서 만든 오픈소스 라이브러리
– GPU(그래픽 카드, Graphic Processing Unit)를 통한 연산을 지원하기 때문에 연산이 복잡한 딥러닝 알고리즘 구현 가능
– 실제 계산은 C++로 구현되었지만, 주된 API는 파이썬으로 이용 가능하기 때문에 쉽게 사용할 수 있고, 계산 속도 또한 빠르다는 장점이 있음

2. 설치

1) Python 3.5 설치

– tensorflow는 Python 3.5 환경에서 지원되므로, Python 3.6아닌 Python 3.5를 설치해야함
– Python 3.5와 pip 설치는 아래의 링크 참고

– Python 3.5 설치 – ‘윈도우 환경에서 Python2와 Python3 함께 사용하기‘ 참고
– pip 설치 – ‘pip를 활용하여 파이썬 패키지 설치하기‘ 참고

2) CUDA 설치

– https://developer.nvidia.com/cuda-downloads 접속

cuda_1

– 자신의 운영체제를 선택한 후 빠른 다운로드를 위해 Installer Type을  local 로 선택하여 다운로드

cuda_2

– 다운로드된 설치 파일을 실행한 뒤 ‘OK” 버튼 클릭

cuda_3

– ‘동의 및 계속’ 클릭

cuda_4

– ‘다음’ 클릭

cuda_5

– ‘NEXT’ 클릭

cuda_6

– ‘다음’ 클릭

cud_7

– ‘닫기’ 클릭

cuda_8

3) CuDNN 설치

– Deep Learning을 위해 nvidia에서 제공하는 추가 라이브러리

– https://developer.nvidia.com/cudnn 접속
– 다운로드시 로그인을 해야 하므로 아이디가 없다면 회원가입부터 진행
– 로그인 후 ‘Download’ 버튼 클릭

cudnn_1

– 다운로드된 압축파일을 실행하여 압축 해제

cudnn_2

– 압축 해제된 폴더에서 폴더 내에 있는 파일들
(\bin\cudnn64_5.dll,
\include\cudnn.h,
\lib\x64\cudnn.lib)을
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\x64
에 복사 붙여넣기 하기

– 단순 반복 작업이므로 본 페이지에서는 \bin\cudnn64_5.dll을 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin로 복사하여 붙여넣는 과정만 설명

– 압축 해제된 폴더에서 bin 폴더를 열고 cudnn64_5.dll 복사하기

cudnn_3

– C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin 으로 이동하여 복사한 파일 붙여넣기

cudnn_4

4) Tensorflow 설치하기

– cmd창 열기

tensor_1

– py -3.5 -m pip install tensorflow-gpu 입력 후 엔터

tensor_2

3. Tensorflow 설치 확인하기

– Pycharm을 열고 python 3.5 버전의 새로운 프로젝트 생성

tfexecute_1

– 새로운 .py 파일을 생성

tfexecute_2

– import tensorflow as tf 입력 후 실행(Ctrl + F9)

tfexecute_3

– 아래와 같은 메세지가 출력된다면 tensorflow가 올바르게 작동하는 것임

tfexecute_4

답글 남기기