Tensor Flow (gpu) 윈도우에서 설치하기

---

순서

1. Cuda 설치

2. cuDNN 설치

3. 파이썬에서 tensorflow-gpu 설치

---

1. Cuda 설치

Cuda 는 Nvidia에서 제공하는 GPU 를 사용하여 병렬처리를 도와주는 모듈이다.

공식 사이트에서 본인에 맞는 설치파일을 다운받아 설치한다.

https://developer.nvidia.com/cuda-downloads

나같은 경우에는 cuDNN 과 버전호환이 잘 맞지않아 9.0버전으로 다운받아 설치하였다.

https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork

설치전 경로를 잘 확인해둔다. (cuDNN 설치시 필요함) 설치파일을 실행하여 다음다음 눌러서 완료

---

2. cuDNN 설치

Cuda 에서 Deep Neural Network을 지원해주는 모듈이다.

https://developer.nvidia.com/cudnn

압축을 해제하면 cuda 폴더안에 bin, include, lib 폴더가 생성된다. 이 폴더를 복사하여서 본인이 설치한 Cuda 경로에 붙여넣기 한다.

나같은 경우에는 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 에 설치되어있다.

---

3. Tensorflow-gpu 설치

윈도우에서 pip은 한계가 있기때문에, anaconda 사용을 아주 강력히 추천한다. 귀찮더라고 꼭 배우시길..

pip install tensorflow-gpu 로 설치한다.

---