TensorFlow-gpu를 설치해봅시다(pycharm)

cuda 10.0
cuDNN 7.6.4
tensorflow-gpu 2.0.0

이렇게 설치할 예정입니다.

<<Nvidia 그래픽카드를 사용하고 있으셔야 cuda가속을 사용하실 수 있습니다.>>

1. cudo 10.0

https://developer.nvidia.com/cuda-10.0-download-archive

여기서 다운로드 받습니다.

설치는 더블클릭하면 알아서 됩니다.

cudo 9.0은 업그레이드도 있으니 받아서 계속 설치합니다.

아나콘다 환경을 사용하실 경우

conda install cudatoolkit==10.0

으로 편리하게 설치하실 수 있습니다.

2. cuDNN 7.5

https://developer.nvidia.com/cudnn

이녀석은 가입을 해야 받을 수 있습니다.

다운로드 하여 압축을 푼 이후

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

여기에 붙여넣기 해서 옮겨줍니다.

마찬가지로 아나콘다 환경을 사용하신다면

conda install cudnn

으로 편리하게 설치하실 수 있습니다.

3. tensorflow-gpu 2.0.0

버전을 굳이 지정한 이유는

이후 버전에서는 tensorflow를 import하는 부분에서 에러가 나는 버그가 있기 때문입니다.

여러 버전을 시도해보았는데 1.10.0 버전이 가장 최신버전으로 오류가 나지 않았습니다(2019.03.24 기준)

2019.08.08 정정합니다 1.14.0 버전이 안정적입니다.

2019.12.02 tensorflow 2.0버전업으로 인하여 업데이트 합니다.

pip install --upgrade tensorflow-gpu==2.0.0

cmd에서 명령어를 이용하여 설치합니다.

이후 pycharm에서

import tensorflow as tf

로 import하면 정상적으로 동작됩니다.

겪었던 여러가지 오류

1. CondaHTTPError: HTTP 000 CONNECTION FAILED for url

openSSL을 설치하면 해결됩니다.

https://slproweb.com/products/Win32OpenSSL.html

위 사이트에서 openssl.exe를 다운받아 설치합니다.

2. return _load(spec) error

텐서플로우 버전을 낮추면 됩니다.

pip uninstall tensorflow-gpu

pip install --upgrade tensorflow-gpu==1.10.0

위 명령어를 이용하여 버전을 1.10.0으로 설치해보세요

3. no cuDNN

cuDNN이 없어서 발생하는 오류입니다.

tensorflow-gpu와 cuda toolkit만 설치할 경우 발생했었습니다.

위 설명을 이용하여 cuDNN을 다운받아 설치하면 해결됩니다.