# 참고 사이트 : 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