pbj0812의 코딩 일기

[TENSORFLOW] WINDOW TENSORFLOW-GPU 버전 설치하기 본문

ComputerLanguage_Program/PYTHON

[TENSORFLOW] WINDOW TENSORFLOW-GPU 버전 설치하기

pbj0812 2018. 10. 30. 00:38

컴퓨터 새로 맞춘지 6개월이 지난 지금에서야 겨우 TENSORFLOW-GPU 버전을 설치하였다...


그만큼 초보자에게 있어 GPU 버전 설치는 힘들다....


CUDA랑 CUDNN 버전을 맞춰야 하고 또 파이썬을 뭘 맞춰야 하고...


그래서 이번에 설치한 김에 복기를 하여 적기로 하였다...


물론... 이건 지금 내 컴퓨터의 상황에 맞춰 설치한 것이기 때문에...


모든 분에게 해당하는 이야기는 아닐지도 모른다...


결국 자신의 컴 상황에 맞춰 설치해야 하기 때문에 그냥 경우의 수를 한 가지 늘린 것이라 보면 좋겠다.




결론은! ANACONDA를 이용하여 설치하였다.


결국 문제는 CUDA랑 CUDNN 버전을 맞추는게 어렵기 때문인데... ANACONDA를 이용하면 자동으로 설치할 수 있다.




1. CUDA를 설치한다.


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


위 경로를 통하여 CUDA를 설치한다. 


나 같은 경우엔 10점대 버전을 설치하였다.




2. ANACONDA를 설치한다.


기존의 PYTHON은 다 삭제하고 ANACONDA를 이용하여 파이썬을 설치하였다.


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


위 경로를 이용하여 ANACONDA를 설치하고 아래 명령어를 통하여 PYTHON 3.5 버전을 설치한다.


(현재는 3.6버전도 지원할 수 있다고 알고 있으나... 다른 분께서 확인해 주시길....)


conda install python=3.5



3. ANACONDA를 이용하여 TENSORFLOW-GPU 버전을 설치한다.


conda install tensorflow-gpu 명령어를 이용하여 설치한다.


이때 필요한 numpy, tensorflow, cudnn 을 설치하기 때문에 편하게 설치할 수 있다.



4. 실행


anaconda prompt 실행 -> python -> import tensorflwo as tf 를 통하여 설치를 확인한다.


ctrl+del를 눌러 작업 환경 창에 들어간 다음 gpu 메모리를 확인한다.

Comments