[Tensorflow] Tensorflow GPU 버전 설치하기

3일의 삽질이 끝났다...

나중에 참고하려고 써보는 포스팅이다. 

버전이 참 중요하다고 새삼느꼈다. 

수많은 버전 충돌로 인한 에러를 충분히 겪고 쓰는 포스팅이 많은 사람들에게 도움이 되길 바라며...

설치 환경 

- Windows 10 / 64 bits 

- Anoconda 4.3 

- Visual Studio 2015

새로 설치한 환경 

- CUDA 8.0 

- cuDNN 6

- Tensorflow 1.4 

설치 순서는 아래와 같다.

아나콘다 - 비주얼 스튜디오(필요시) - CUDA - CuDNN - Tensorflow 

* 참고로 Tensorflow 1.5 버전은 CUDA 9.0 이 필요하고,

* Tensorflow 1.3 버전 부터는 cuDNN 6 버전이 필요하다. 

* Tensorflow 1.5 버전이 릴리즈 되었다고 한다

참고자료 : https://tensorflow.blog/tag/tensorflow/

1. CUDA 8.0 설치하기

https://developer.nvidia.com/cuda-80-ga2-download-archive

2. cuDNN 6 설치하기 (회원가입)

https://developer.nvidia.com/rdp/cudnn-download

cuDNN v6.0 Library for Window 10 을 설치하였다. 

다운로드 한 뒤, 압축을 풀고 폴더 내부에 있는 bin, include, lib 파일을 복사하여 

아래 경로에 붙여넣어 준다.

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

* 참고로 CUDA 8.0 버전에 cuDNN 5.1 버전을 설치하라는 이야기가 있어서 텐서플로우를 설치하고 임포트 했더니 Could not find 'chdnn64_6.dll' 이라는 아래와 같은 에러가 났다. 

텐서플로우 1.2 버전 까지는 cuDNN 5.1을 사용하였는데 1.3 이상부터는 cuDNN 6 버전을 사용하여야 한다고 한다. 난 텐서플로우 1.4 버전이니까 cuDNN 6 버전을 다시 설치하여 에러를 해결하였다. 

에러 참조 : http://boysboy3.tistory.com/115

Traceback (most recent call last):

  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\platform\self_check.py", line 87, in preload_check

    ctypes.WinDLL(build_info.cudnn_dll_name)

  File "C:\Anaconda3\envs\tensorflow\lib\ctypes\__init__.py", line 351, in __init__

    self._handle = _dlopen(self._name, mode)

OSError: [WinError 126] 지정된 모듈을 찾을 수 없습니다

During handling of the above exception, another exception occurred:

Traceback (most recent call last):

  File "<stdin>", line 1, in <module>

  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\__init__.py", line 24, in <module>

    from tensorflow.python import *

  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>

    from tensorflow.python import pywrap_tensorflow

  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>

    self_check.preload_check()

  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\platform\self_check.py", line 97, in preload_check

    % (build_info.cudnn_dll_name, build_info.cudnn_version_number))

ImportError: Could not find 'cudnn64_6.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Note that installing cuDNN is a separate step from installing CUDA, and this DLL is often found in a different directory from the CUDA DLLs. You may install the necessary DLL by downloading cuDNN 6 from this URL: https://developer.nvidia.com/cudnn

3. 아나콘다(Python 3.6) 버전 설치 후 텐서플로우 설치 

아나콘다 설치는 아래 링크에서 하고,

https://www.anaconda.com/download/

설치 방법은 블로그 포스팅의 1번 부분까지 참고하길 바란다.

http://eehoeskrap.tistory.com/214?category=629855

먼저 Anaconda Prompt 를 열어준다. 

tensorflow 라는 이름을 가진 python 버전 3.5의 가상환경을 생성한다.

conda create -n tensorflow python=3.5 

가상환경 활성화

activate tensorflow

텐서플로우 설치하기

pip install --ignore-installed --upgrade http://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl

사이트 참고 : https://pypi.python.org/pypi/tensorflow/1.4.0

파이썬을 이용하여 테스트 (한줄씩)

python

import tensorflow as tf

hello = tf.constant('Hello, Tensorflow!')

sess = tf.Session()

print(sess.run(hello))