출처: http://jaejunyoo.blogspot.com/2017/02/start-tensorflow-gpu-window-10.html
https://www.tensorflow.org/install/install_windows
이제 막 딥 러닝을 공부해보려고 시작한 사람입니다. TensorFlow 설치를 할 때마다 tensorflow.org의 설치 가이드(https://www.tensorflow.org/install/install_windows) 대로 해도 설치가 잘 안되고 계속 혼동이 되서 설치에 성공한 과정의 기록을 남깁니다. 저는 Windows 10, 64 bit 환경이라 다른 분들과는 환경이 다를 수 있습니다.
2017년 7월 24일 기준의 설치 방법입니다.
1. CUDA Toolkit 8.0 설치
https://developer.nvidia.com/cuda-downloads 에서 Windows용 CUDA Toolkit 8.0을 받아서 설치합니다.
exe(network) 보다는 exe(local)이 설치가 빠릅니다.
2. cuDNN v5.1 설치
https://developer.nvidia.com/cudnn 에서 cuDNN v5.1을 받아서 설치합니다.
cuDNN v6.0도 있는데 v6.0을 쓰면 설치가 안됩니다.
cuDNN v5.1을 받아서 압축을 풀면 cuda 밑에 bin, include, lib 디렉토리가 생기는데
이 3가지 디렉토리들을 CUDA 설치 디렉토리에 복사해줍니다.
CUDA 설치 디렉토리는 C:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v8.0 입니다.
32bit에서는 아마도 C:\Program Files (x86)\NVIDIA GPU Computing Tookit\CUDA\v8.0일 겁니다.
3. Anaconda 설치
https://www.continuum.io/downloads 에서 Anaconda 4.4.0 For Windows를 받아서 설치합니다.
버전이 여러가지가 있는데 Python 3.6 Version의 64 bit를 다운 받습니다.
설치할 때는 Just Me와 All Users 옵션 중에서 Just Me를 선택합니다.
All Users로 설치해보니 동작이 이상해서 다시 지웠습니다.
(아마도 관리자 권한으로 수행시켜야 제대로 동작하는 것 같습니다.)
4. Tensorflow-gpu 설치
Anaconda Prompt를 수행하고 Anaconda Prompt에서 다음과 같이 타이핑 해서 설치합니다.
Anaconda Prompt는 Anaconda 3 (64bit) -> Anaconda Prompt 아이콘을 클릭해서 수행합니다.
원래 Tensorflow.org에 나온 설치 명령어는 다음과 같은데, 이러면 Python 3.5가 필요합니다.
cp35부분을 cp36으로 바꾸니 python 3.6에서도 TensorFlow가 제대로 동작하는 것 같습니다.
5. 설치 검증
Tensorflow.org의 Validate your installation 부분에서 나온 절차대로 Anaconda Prompt를 띄우고
python을 수행한 다음 python shell 에서 Hello TensorFlow!를 출력하는 문장을 따라해보면
문제없이 동작하는 것을 보실 수 있습니다.