본문 바로가기
Developer/Machine Learning

#2. TensorFlow 설치 - GPU

by Doony 2017. 11. 13.

본 포스팅은 인프런에서 제공되는 "모두를 위한 딥러닝 - 기본적인 머신러닝과 딥러닝 강좌"를 참고하여 만들어졌습니다. 

인프런 URL: https://www.inflearn.com/

텐서플로우는 가장 많이 활용되고 있는 머신러닝 오픈소스 라이브러리이다. 그만큼 자료도 많고 배우기도 용이하다고 하는데... 

TensorFlow is an open source software library for numerical computation using data flow graphs. With Python!

Data flow graph 는 순서도 같은 것. Node(점)과 이를 잇는 Edge(선)으로 구성됨. 

Node: mathematical operation

Edge: Data arrays (Tensors)


일단 설치부터 해야겠지?


  Tensorflow 설치하기

먼저 친절한 설명은 https://www.tensorflow.org/install/install_windows 공식홈페이지에서 참조할 수 있다.

CPU보다 GPU로 연산하는 게 빨라, 두가지 종류로 텐서플로우를 설치할 수 있는데, 나는 지원되는 그래픽카드를 사용해서 GPU버전으로 설치를 진행했다.




3. cuDNN 6 다운 후 환경변수 등록

https://developer.nvidia.com/cudnn

다운받으면, 폴더가 나올 것이다. cuDNN 라이브러리로 활용될 것이므로, 적당한 폴더에 놔두고 환경변수에 해당 경로를 설정해주자.

(환경변수 설정은 윈도우+pause-break키 --> 고급시스템설정 --> 환경변수 --> path 수정 --> 새로만들기.)


4. 텐서플로우 설치

명령프롬프트창을 킨다.

pip3 install --upgrade tensorflow-gpu 



#1. 머신러닝의 정의 Machine Learning

공감은 큰 힘이 됩니다.

두니월드로 놀러오세요.




댓글