Tensorflow GPU 환경구축

Tensorflow GPU환경 구축을 위한 CUDA Toolkit, cuDNN v5.1 두 가지 프로그램을 설치해야 한다.

1. CUDA Toolkit을 설치한다.

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

2. cuDNN 설치한다.

https://developer.nvidia.com/cudnn

cuDNN의 경우 Nvidia Developer사이트의 회원 가입이 필요하므로 가입하면 된다.

주의 할점은 윈도우 10환경에서는 Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0 즉, 5.1버전으로 다운로드 받아야 오류가 발생하지 않는다.

3. 텐서플로우 GPU버전 설치한다.

pip install --upgrade tensorflow-gpu 

4. anaconda prompt에서 방금 만든 환경으로 접속한다.

activate tensorflow