TensorFlow 설치

머신러닝이나 딥러닝을 하면 많이 쓰는 라이브러리 중에 하나인  TensorFlow 설치 방법에 대해서 설명하겠습니다.

저는 아나콘다를 이용해서 설치하는 방법을 설명하겠습니다.

 

TensorFlow 설치를 하기 전에

TensorFlow를 설치 전에 준비해야할 것이 있습니다. 바로 아나콘다와 TensorFlow-gpu를 사용하는 경우 cuda 를 설치해야 합니다.

 

아나콘다 설치방법은 일전에 포스팅을 해놔서 참고하시면 되겠습니다.

https://seong6496.tistory.com/22

[Python] 아나콘다 설치하기

 

만약 TensorFlow-gpu 를 사용해야 한다면 cuda 를 설치해야 합니다.

cuda 도 일전에 포스팅을 해놨습니다.

https://seong6496.tistory.com/23?category=889148

[Python] CUDA 설치 및 버전 확인

---

TensorFlow 설치

아나콘다와 cuda 를 깔았다면 이제 TensorFlow 를 설치하는 일만 남았습니다.

아나콘다를 이용해서 TensorFlow 를 설치해보겠습니다.

 

먼저 윈도우 시작에서 아나콘다 펼쳐보면 Anaconda Powershell Prompt 를 관리자권한으로 들어가면 도스창이 하나 뜹니다. 여기서 TensorFlow를 설치를 하면 되는데 pip 이 최신버전이어야 합니다.  다음을 입력해서 pip을 최신버전으로 바꿔줍니다.

python -m pip install --upgrade pip

이미 최신 버전이면 다음과 같이 출력이 됩니다. 그대로 사용하시면 됩니다.

Requirement already up-to-date:

이렇게 pip 을 최신버전으로 바꾸고 TensorFlow를 다음과 같이 입력해 설치합니다.

pip install tensorflow

 

 

1.15.0 버전과 2.1.0 버전부터는 CPU,GPU 패키지가 통합되어있습니다. 그래서 NVIDIA 계열의 그래픽카드를 가지고 계신다면 GPU만 따로 설치하실 필요가 없습니다. 즉, CPU 계산을 하지 못하게 되는 일이 없어서 가상환경을 따로 만들필요가 없습니다. 코드 구성도 같아서 같이 쓰셔도 무방합니다. 다만, CPU만 사용해야 한다면 tensorflow 대신에 tensorflow-cpu 라 입력하시면 되겠습니다. GPU 만 사용하겠으면 tensorflow-gpu가 되겠습니다.

 

 

AVX 가 지원되지 않는 경우는 1.5.0 버전을 설치하셔야 합니다. 

pip install tensorflow==1.5.0

---

설치 확인

설치가 다 되었다면 잘 되는지 확인해봅시다.

Anaconda Powersherll Pormpt 에서 python 이라 명령하면 

python을 쓸수 있게끔 >>> 이런 표시가 나옵니다. 

1.x 버전인 경우

import tensorflow as tf
hello = tf.constant('Hello,TensorFlow')
sess = tf.Session()
print(sess.run(hello))

2.x 버전인 경우 tf.session 이 삭제되어서 다른 방식으로 해야합니다.

import tensorflow as tf
hello = tf.constant('Hello,TensorFlow')
hello.numpy().decode('utf-8')

 

정상적으로 설치가 되었다면 'Hello,TensorFlow' 가 나올겁니다!

 

---

CPU AVX 지원 확인

아래에 링크를 한 홈페이지에 가면 coreinfo 라는 프로그램을 다운 받으실 수 있습니다.

coreinfo 홈페이지

Coreinfo - Windows Sysinternals

위 홈페이지에 가시면 coreinfo라는 zip 파일을 받을 수 있습니다. 압축을 풀어서 cmd 에서 압축을 푼 해당폴더로 가서 coreinfo라고 치시면 AVX 이 지원되는지 확인하실 수 있습니다.