2018-03-06

Windows에 TensorFlow GPU 버전 설치하기

# 참고 사이트 : https://brunch.co.kr/@gnugeun/20

# Windows 10 64bit 영문판, nvidia GeForce940MX, tensorflow-gpu 설치 기준
- 관건은 CUDA, cuDNN과 python의 버전을 맞추는 것임.
- 2018년 3월 6일 기준으로 CUDA 9.0, cuDNN 7.1, python 3.6.4 에서 이상없이 작동됨.
- 만일 gpu 버전이 아닌 경우에는 STEP 3로 바로 넘어가도 됨.

STEP 1. CUDA 9.0 설치
- https://developer.nvidia.com/cuda-80-ga2-download-archive

STEP 2. cuDNN 7.1 설치 : 7.1 for CUDA 9.1도 상관없음
- https://developer.nvidia.com/rdp/cudnn-download
- nvidia 회원가입 필요
- 단순히 압축을 푸는 용도이므로 CUDA 내 해당되는 디렉토리에 복사해 넣음
- 또는 path 설정시 cuDNN 폴더를 지정

STEP 3. python 3.6 64bit 설치
- https://www.python.org/downloads/windows/
- 최신 안정버전인 3.6.4 windows x86-64 executable installer 다운로드
- 설치시 path 업데이트에 꼭 체크할 것 : Add Python 3.6 to PATH

STEP 4. TensorFlow 설치
- cmd를 관리자 권한으로 실행
- pip upgrade
  > python -m pip install --upgrade pip
   (최신 버전에서는 이미 업그레이드 되어 있어 설치되는 건 없다)
- TensorFlow 설치
  > pip install --upgrade tensorflow-gpu
- 만일 gpu 버전이 아니라면 cpu 버전을 설치하면 됨
  > pip install --upgrade tensorflow

Further. 보너스 : Jupyter Notebook 설치
  > pip install Jupyter

# TensorFlow 설치 : Anaconda판
- https://www.anaconda.com/download/
- 설치한 python 버전에 맞는 버전 설치
- 설치한 후에는 Anaconda prompt를 관리자 권한으로 실행
- pip 업그레이드
  (base)> python -m pip install --upgrade pip
- Conda 환경 만들기
  (base)> conda create -n tensorflow python=3.6.4
         (중간에 Proceed([y]/n? 나오면 y 선택)
- TensorFlow 설치
  (base)> activate tensorflow
         (프롬포트가 tensorflow로 바뀜)
  (tensorflow)> pip install tensorflow-gpu

### 2018. 4. 10 추가
현재 CNN이 작동되지 않는데, 다른 건 다 되면서 왜 안되는지 모르겠다.

No comments:

Post a Comment