[인공지능] TensorFlow 설치하기

참조

• https://www.tensorflow.org/install/pip?hl=ko
• https://github.com/tensorflow/tensorflow/releases

소개

• 안녕하세요. 오늘은 TensorFlow 설치하는 방법에 대해서 알려 드리려고 합니다.
• 회사에서 TensorFlow를 설치해서 작업할 내용이 있어서 기록할 겸 포스팅 하게 되었습니다.

하드웨어 및 OS 스펙

• OS : Windows 10
• CPU : Intel(R) Core(TM) i9-9900K CPU @ 3.60GHz 3.60 GHz
• GPU : NVIDIA GeForce 1050 Ti
• RAM : 32GB

TensorFlow 설치 순서

1. TensorFlow 버전 선정
2. Python 설치
3. Cuda, cuDNN 설치

1. TensorFlow 버전 설정

• 현재 기준으로 Tensorflow 는 2.8.0 버전까지 Release 되었습니다.
• 저는 TensorFlow 가장 최신 버전인 2.8.0 버전을 설치할 예정입니다.

2. 파이썬 버전 선정

• 현재 기준으로 Tensorflow 는 2.8.0 버전까지 Release 되었습니다.
• 하지만, https://www.tensorflow.org 사이트에서는 Tesnsorflow 2.7.0 버전의 CPU, GPU 테스트 빌드 구성의 버전까지만 나와있어서 해당 버전을 기준으로 Python 버전 설치 진행합니다.
• 현재 Tensorflow 2.7.0 기준으로 Python 버전은 3.7 ~ 3.9 버전을 요구하고 있기 때문에, 그중 가장 최신 버전인 3.9 버전을 설치합니다.

3. CPU, GPU 테스트 빌드 구성 환경 확인

• 아래와 같이 현재 Python 버전은 3.7 ~ 3.9 버전을 요구하고 있습니다.

4. 파이썬 홈페이지 접속하여 설치 진행

• 현재 날짜 기준으로, 3.9 버전에서는 3.9.10 버전이 가장 최신 버전이면서 안정적인 릴리즈 버전입니다.
• 3.9.10 버전을 설치 진행합니다.
• https://www.python.org/downloads/windows/ 위 링크를 통해서 설치 가능합니다.

5. 파이썬 설치 완료

6. TensorFlow pip 설치

• 다음 명령어를 통해 TensorFlow pip 설치를 진행합니다.

pip3 install --user --upgrade tensorflow  # install in $HOME

7. 설치 확인

• pip show tensorflow 명령어를 통해 tensroflow 설치 유무 확인하였습니다.
• 2.8.0 버전의 tensorflow가 정상적으로 설치된 것을 확인할 수 있습니다.

C:\Users\bh.cho>pip show tensorflow
Name: tensorflow
Version: 2.8.0
Summary: TensorFlow is an open source machine learning framework for everyone.
Home-page: https://www.tensorflow.org/
Author: Google Inc.
Author-email: packages@tensorflow.org
License: Apache 2.0
Location: c:\users\bh.cho\appdata\roaming\python\python39\site-packages
Requires: absl-py, tf-estimator-nightly, protobuf, grpcio, flatbuffers, keras-preprocessing, astunparse, google-pasta, six, typing-extensions, tensorboard, h5py, numpy, libclang, tensorflow-io-gcs-filesystem, setuptools, wrapt, termcolor, keras, gast, opt-einsum
Required-by: