[Windows10] tensorflow GPU 에러 해결 & 설치하기

윈도우에 텐서플로우 GPU버전을 처음 설치해보는 지라 이런 에러를 접하니 적지않게 당황스러웠다.

그래서 구글링을 해본 결과 cuDNN설치를 제대로 안 한 것이였다.. 이 에러 해결 방법은 밑에 서술하여 놓았다.

python 버전 3 기반으로 설치하였습니다.

1. https://developer.nvidia.com/cuda-downloads 접속을 하여 자신에게 맞는 운영체제를 선택합니다.

저는 Base installer로 설치하였습니다.

2. 그 다음 https://developer.nvidia.com/cudnn 접속하여 cuDNN을 설치합니다. 버전은 5.1 버전으로 설치해야 됩니다.

그럼 알집 파일이 다운로드가 될텐데, 압축해제를

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

 이 경로에 해줍니다.

만약 이 경로에 폴더가 없다면 1번 설치를 다시 해 보시길 바랍니다.

3. 이제 파이썬이 설치되어있다는 가정 하에 cmd창을 열어줍니다.

pip3 install --upgrade tensorflow     //텐서플로우 cpu버전 설치 pip3 install --upgrade tensorflow-gpu //텐서플로우 gpu버전 설치

커맨드 라인에 입력해 줍니다.

파이썬 버전3이 설치되어 있으면 자동으로 tensorflow가 설치 될 것입니다.

4. 자 그럼 이제 텐서플로우가 잘 설치되었는지 확인을 해 볼 차례입니다.

커맨드 라인에 python을 입력하여서 파이썬에 접속하시고

>>> import tensorflow as tf >>> hello = tf.constant('Hello, TensorFlow!') >>> sess = tf.Session() >>> print(sess.run(hello))

요 명령어를 입력하셔서 오류 없이 b'Hello, TensorFlow!' 또는  'Hello, TensorFlow!' 가 나온다면 라이브러리가 잘 설치된 것입니다.