윈도우 10에 텐서플로우 설치하기 (GPU 버전)

설치순서를 간략히 정리하면 아래와 같다.

1. 파이썬 또는 아나콘다 배포판 설치

2. CUDA Toolkit 설치

3. cuDNN SDK 설치

4. Tensorflow GPU 버전 설치

---

1. 파이썬 또는 아나콘다(Anaconda) 배포판 설치 

텐서플로우 GPU 버전을 설치하기 위해서는 파이썬이 설치되어 있어야 하며, 보통 파이썬을 포함하고 있는 아나콘다라는 배포판을 설치한다. 관련 내용은 아래 사이트를 참고하시길 바란다.

윈도우 10에 텐서플로우 설치하기 (CPU 버전) <- 1. 아나콘다 배포판 설치만 확인

설치 전에 텐서플로우 사이트에서 지원되는 하드웨어, 소프트웨어 정보를 확인하고 자신에게 맞는 버전을 설치하도록 하자.

무조건 최신버전을 설치하는 경우 지원이 안되는 경우가 종종 있기 마련이다.

1. CUDA Toolkit 설치

글을 작성하고 있는 시점의 Tensorflow 설치를 위해서는 CUDA 9.0을 설치해야한다. NVIDIA 사이트에서 해당 파일을 다운로드한다.

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

Download Now를 실행하면 CUDA 10.0 버전이므로 9.0 버전을 설치한다.

프로그램을 다운로드하여 설치프로그램을 실행한다.

설치시작과 함께 호환성 검사를 실행한다.

설치 옵션에서 주의할 부분이 있다. 빠른 설치를 하는 경우 불필요한 드라이버까지 설치하면서 호환성문제로 설치를 실패하는 경우가 많으니 반드시 사용자 정의 설치를 선택하도록 한다.

아래 두가지만 체크하여 실행한다.

설치목록이 표시된다. 정상적으로 설치되었다.

2. cuDNN SDK 설치

아래 사이트 접속하여 설치프로그램을 다운로드 한다.

https://developer.nvidia.com/cudnn

다운로드 페이지에서 시스템 환경에 맞는 버전을 선택한다.

설치 방법은 다운로드 받은 zip 파일(cudnn-9.0-windows10-x64-v7.3.1.20.zip)의 압축을 풀어 CUDA 설치 디렉토리에 복사하면 된다.

Default로 설치한 경우 설치경로는 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 이며, 아래는 설치된 모습이다.

4. TensorFlow GPU 버전 설치

>> pip install --upgrade tensorflow-gpu

설치 후 텐서플로 코드를 실행하여 동작을 확인한다.

>> import tensorflow as tf

>> hello = tf.constant(;hello tensorflow gpu')

>> sess = tf.Session()

>> print(sess.run(hello)