[Tensorflow] tensorflow-gpu 설치

1. CUDA Toolkit 설치

- CUDA Toolkit 9.0을 다운로드 했습니다. cuDNN 과의 호환 때문에 버전을 신경써서 설치해야 합니다. (그림 1)을 보면 알 수 있듯이, 자신의 컴퓨터 사양과 맞는 것을 클릭하고 다운로드 받습니다 [다운로드 링크]. 

​

​​

​

2. cuDNN 설치

7.0.5 버전을 다운로드 받았습니다 [다운로드 링크].

 

( 그림2 ) cuDNN 설치 버전

​

설치 후에는 압축파일에서 압축 해제 후 cuda 폴더에 있는 bin, include, lib 폴더를 NVIDIA GPU Toolkit 폴더 밑에 CUDA/v9.0 폴더에 옮겨줍니다. ( 그림3 ) 폴더에 파일들을 ( 그림4 ) 폴더로 옮겨준다.

​

( 그림3 ) cuDNN 폴더를 압축해제 하면 위와 같은 폴더가 생성된다.

​

( 그림4 ) CUDA Toolkit을 설치하면 다음과 같은 폴더가 생성된다.

​

 

3. 환경변수 설정

- 여기 부분에서 삽질을 한 것 같다.

이게 처음에 다른 글들 보고 하다가 TOOLKIT 10점대 버전 설치하고 8점대 버전 설치하고 최종적으로 9점대 버전을 설치했다. 그러다보니 밑에 CUDA_PATH_V8_0, CUDA_PATH_V10_0 등이 생기면서 경로를 못 잡는 상황이 발생했었다.

( ImportError: DLL load failed ) 이런 에러가 발생한다면 환경변수 설정을 잘 못해서 발생하는 에러이다.

​

그냥 딱 더도 말고 덜도 말고 CUDA_PATH와 CUDA_PATH_V9_0만 확인하면 된다.

괜히 사용자 변수 편집 버튼 눌러서 이상한 거 추가하다 더 안된 것 같다.

 

( 그림5 ) 환경 변수 설정

환경 변수 설정은 밑에 페이지에 잘 설명되어 있다.

https://m.blog.naver.com/chandong83/221112939772

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

​

​

4. Anaconda 설치

밑의 URL에서 Python 3점대 버전을 설치해줍니다. 가급적 Anaconda를 설치해서 환경을 분리해주는 것이 좋습니다.

나중에 환경설정이 엉켜 감당할 수 없습니다.

​​anaconda prompt 창에 들어가 아래의 명령어를 입력합니다.

conda create -n cuda pip python=3.5 /*환경 만들기(python 3.5) -> y/n 나오면 y 누를것.*/
activate cuda /*환경 활성화*/


python -m pip install --upgrade pip /*일단 pip 를 업그레이드 시켜준다.*/
pip install --ignore-installed --upgrade tensorflow-gpu==1.5

 

*** 중요 ***

tensorflow-gpu 1.5 버전 설치해야 한다.

최신 버전 설치하면 DLL 에러 발생한다!!!

​

​

5. jupyter notebook 설치

pip3 install jupyter

jupyter notebook 실행 한 후에 테스트 하면 끝나게 된다.

​

​

​

6. 테스트

=> 밑의 코드를 실행시키고 에러가 발생하지 않는다면 설치에 성공한 것이다.

$ python
…
>>> import tensorflow as tf
>>> hello = tf.constant(‘Hello, TensorFlow!’)
>>> sess = tf.Session()
>>> sess.run(hello)
b’Hello, TensorFlow!’
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> sess.run(a + b)
42
>>>

​

Reference

[1] https://devyurim.github.io/python/tensorflow/2018/04/30/tensorflow-1.html

[2] http://solarisailab.com/archives/1581

[3] https://m.blog.naver.com/chandong83/221112939772