설치는 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'을 누르면 실행 결과가
아래와 같이 노출된다.
설치와 테스트가 완료되었다.
' 장난감 > Tensorflow 관련' 카테고리의 다른 글
Visual Studio Code에서 Jupyter을 사용하려 한다. (2) | 2017.06.06 |
---|---|
Visual Studio Code에서 Python 사용하기 (0) | 2017.05.13 |
TensorFlow를 윈도우에 설치하기, Python pip설치와 Docker로 간단히 설치하자. (0) | 2017.05.06 |