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

텐서플로 GPU 버전 설치순서를 간략히 정리해보겠습니다.

1. 파이썬 또는 아나콘다 배포판 설치
2. CUDA Toolkit 설치
3. cuDNN SDK 설치
4. Tensorflow GPU 버전 설치


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

텐서플로우 설치를 위해서는 파이썬이 설치되어 있어야 하며, 보통 파이썬을 포함하고 있는 아나콘다라는 배포판을 설치합니다. 아나콘다(Anaconda) 설치는 이전 글인 ‘윈도우 10에 텐서플로우 설치하기 (CPU 버전)’를 참고하시길 바랍니다.

파이썬이 설치되었으면 CUDA Toolkit, cuDNN, Tensorflow gpu 버전순으로 설치합니다.

설치 전 텐서플로우 사이트에 방문하여 하드웨어, 소프트웨어 호환성 정보를 확인합시다.  호환성 문제가 없다면 최신버전을 설치하는 것이 좋겠습니다. CUDA는 9.0, cuDNN은 7.2 이상 요구된다고 명시되어 있습니다.

2. CUDA Toolkit 설치

위의 이미지를 보시면 아시겠지만 글을 작성하고 있는 시점에서는 CUDA 9.0을 지원합니다. NVIDIA 사이트에서 해당 파일을 다운로드합니다. (https://developer.nvidia.com/cuda-zone)

바로 다운로드를 받게되면 CUDA 10.0 버전이므로 이전버전을 찾아 9.0 버전을 설치합니다. CUDA 10.0이 지원되는 시점에는 10.0을 그냥 받으시면 됩니다.

프로그램을 다운로드하여 설치프로그램을 실행하고 호환성검사가 실행됩니다.

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

아래 두가지만 체크합니다.

설치목록이 표시되며 정상적으로 설치되었음을 확인할 수 있습니다.

3. cuDNN SDK 설치

이번에는 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))

정상적으로 동작됨을 확인할 수 있습니다.

4 thoughts on “윈도우 10에 텐서플로우 GPU 버전 설치하기”

  1. 안녕하세요.
    위의 글대로 따라했는데 “Adding visible gpu devices: 0 이후로 진행이 안되네요.
    어디를 살펴봐야 잘못된 것을 고칠 수 있을까요?

  2. 안녕하세요, 설치도중 궁금한 게 있어 댓글 남깁니다.
    session을 생성하면 생성만 될 뿐 위 사진처럼 로그 메시지가 안 뜨는데 설치가 잘못 된 걸까요??

    • 안녕하세요. 답변이 늦었습니다.

      질문은 세션생성 후 진행되지 않는다는 의미인 것 같은데
      그렇다면 윗 분과 비슷한 case일 것 같습니다.

      설치 중 gpu 드라이버와 cuda 드라이버의 호환성 문제가 많은 편이구요.
      확실한 건 아니고 그렇게 추측됩니다.

Leave a Comment