[Tensorflow]텐서플로우 설치하기

일단 가장 먼저 해야하는것은 내가 gpu를 쓸것인가 쓰지 않을것인가 이다.

gpu를 쓰려면 일단 gpu를 사용할 수 있는 라이브러리가 선행 설치가 되야 될것이다.

nvidia의 cuda와 머신러닝 라이브러리 cudnn이 설치되야한다.

gpu를 쓰지 않는다면(혹시 못쓴다면) 이 과정은 생략해도 좋다.

일단 gpu를 쓴다는 가정하에서 진행해 보도록하자.

cuda설치

https://developer.nvidia.com/cuda-downloads에 가서 쿠다를 설치한다.

사실 뭐 특별할건 없는데 가급적이면 경로는 건드리지 말자.

쿠다를 그냥 설치하면해당경로에 설치된다.

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

이경로는 중요한데 왜냐하면 cudnn을 설치해야하기 때문이다.

cudnn설치

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

위에서 다운을 받는다.

다운을 받으면 zip으로 받는다.

위와 같이 세개의 폴더가 존재한다.

이 폴더의 내용물을 각각 cuda의 설치경로에 가서 동일 이름안에 넣어주면 준비는 끝났다.

tensorflow 사용하기

사실 그냥 설치해도 되는데 아마 윈도우에서는 잘 안될것이다.

리눅스와 맥은 그냥 설치가 되므로 그냥 쓰면 된다.

윈도우에서 잘 안되는 이유는 윈도우만 지원하는 버전이 3.5가 끝이기 때문이다.

따라서(사실 그 이유때문만은 아니지만) 아나콘다에서 가상환경으로 작업해주는게 더 낫긴하다.

그래서 보통은 아나콘다를 사용한다.

아나콘다를 사용하는 방법은 여기를 참고하자.

그 다음은 간단하다 환경을 activate시킨후에 pip로 설치하면 된다.

gpu 사용 -  pip install tensorflow

gpu 미사용 - pip install tensorflow-gpu

그리고 기다리면 설치가 될것이다.

이제 설치가 끝났으면 python에 들어가서 import tensorflow를 했을때 아무 오류가 나지 않으면 성공한 것이다.

참고로 gpu 사용버전이나 미사용버전이나 import tensorflow로 명령어는 똑같다.