TensorFlow를 윈도우에 설치하기, GPU 버젼 설치하기.

2017. 5. 13. 02:56 장난감/Tensorflow 관련


설치는 64bit 환경에서 진행되었다.



1. CUDA / cuDNN 설치

    • TensorFlow사이트에서 GPU 지원 설치 확인한다.
         https://www.tensorflow.org/install/install_windows

    • GPU지원과 관련된 내용 중 

         'CUDA 8.0 , cuDNN 5.1' 를 다운로드 한다.


    • CUDA 설치 후 cuDNN내 bin디렉토리 하위의 cudnn64_5.dll 파일을

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

         로 복사한다.


    • cudnn64_5.dll을 복사하지 않으면, 

         "ImportError: DLL load failed: 지정된 모듈을 찾을 수 없습니다."라는

         반갑지 않은 메시지를 만나게 된다.


    • Visual Studio 2015  또는 Visual Studio 2015 재배포 가능 패키지가 설치되지 

        않았다면, MS사이트에서 다운받아 설치한다.




2. tensorflow 설치

   

    • 파이썬 3.5.x 설치본을 다운로드한다.

         https://www.python.org/downloads/release/python-350/


    • 파이썬에 TensorFlow 모듈을 설치한다.

         pip3 install --upgrade tensorflow-gpu 




3. atom Editor 설치

    • python 에디터 중 atom을 사용해서 실행을 해보도록하자.

        https://atom.io/

        사이트에서 설치 파일을 다운로드 후 설치한다.


    • Settings > Install에서 'python' 패키지를 검색하여

        'atom-python-run'을 설치한다.

        


    • TensorFlow 사이트의 GPU예제를 테스트 해보도록 한다.

   파이썬 파일을 만들어 타이핑 후 'F5 or F6'을 누르면 실행 결과가 

   아래와 같이 노출된다.



설치와 테스트가 완료되었다.