개발_이야기

Do 'tensorflow' in 'win10'

조규현15 2016. 12. 1. 22:12
반응형

windows환경에서는 tensorflow를 사용하기 위해서는 docker(vm)에서 linux로 돌려야했지만

tensorflow 0.12부터 직접 지원을 한다.

https://developers.googleblog.com/2016/11/tensorflow-0-12-adds-support-for-windows.html


조건은 python 3.5 64bit으로 pip 명령으로 설치가 가능하다.

gpu를 사용하기 위해서는 tensorflow-gpu로 설치하면 된다.

그리고 cuda 8.0를 설치한다(설치가 간편하니 생략)

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


마지막으로 cuDNN를 설치하면 된다.

'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0'경로의 bin, include, lib에 파일을 추가한다.

https://developer.nvidia.com/cudnn


GTX 970에서 mnist 예제(https://gist.github.com/haje01/202ac276bace4b25dd3f)가 엄청 빠르게 끝난다.


God Google (*▽)ノ♡ 

반응형