SW/리눅스

CentOS 7 : TensorFlow 설치 방법, 예제, 명령어

얇은생각 2021. 8. 27. 07:30
반응형

텐서플로(TensorFlow)는 구글이 개발한 머신러닝 모델을 구축하기 위한 자유-오픈 소스 플랫폼이다. 트위터, 페이팔, 인텔, 레노버, 에어버스 등 수많은 기관에서 사용하고 있다.

이 튜토리얼에서는 CentOS 7에 TensorFlow를 설치하는 방법에 대해 설명합니다.

TensorFlow는 파이썬 가상 환경, 도커 컨테이너 또는 아나콘다와 함께 시스템 전체에 설치할 수 있습니다.

 

 

CentOS 7 : TensorFlow 설치 방법, 예제, 명령어

 

 

CentOS에 TensorFlow 설치

TensorFlow는 Python 2와 3을 모두 지원합니다.

Python 3을 사용하고 가상 환경 내에 TensorFlow를 설치할 예정입니다. 이렇게 하면 단일 컴퓨터에 여러 개의 분리된 Python 환경을 구축하고 다른 프로젝트에 영향을 줄 염려 없이 프로젝트별로 특정 버전의 모듈을 설치할 수 있습니다.

 

 

 

1. Python 3 설치

소프트웨어 컬렉션(SCL) 저장소에서 Python 3.6을 설치합니다.

CentOS 7은 CentOS 기본 시스템의 중요한 부분인 Python 2.7.5와 함께 제공됩니다. SCL을 사용하면 기본 python v2.7.5와 함께 최신 버전의 python 3.x를 설치할 수 있으므로 yum과 같은 시스템 도구가 계속 제대로 작동할 수 있습니다.

리포지토리를 활성화하려면 SCL 릴리스 파일을 설치합니다.

완료되면 다음 명령을 실행하여 Python 3.6을 설치합니다.

sudo yum install centos-release-scl
sudo yum install rh-python36

 

 

이제 TensorFlow 프로젝트를 위한 가상 환경을 만들 준비가 되었습니다.

 

 

 

2. 가상 환경 생성

Python 3.6부터는 venv 모듈을 사용하는 것이 좋습니다.

Python 3.6에 액세스하려면 scl 도구를 사용하여 새 셸 인스턴스를 실행해야 합니다.

scl enable rh-python36 bash

 

 

TensorFlow 프로젝트를 저장할 디렉토리로 이동합니다. 홈 디렉토리 또는 사용자가 읽기 및 쓰기 권한을 가지고 있는 다른 디렉토리일 수 있습니다.

TensorFlow 프로젝트에 대한 새 디렉토리를 생성하고 cd로 이동합니다.

mkdir tensorflow_project
cd tensorflow_project

 

 

디렉토리에서 다음 명령을 실행하여 가상 환경을 생성합니다.

python3 -m venv venv

 

 

위의 명령은 venv라는 이름의 디렉토리를 생성하는데, 이 디렉토리에는 파이썬 바이너리, Pip 패키지 관리자, 표준 파이썬 라이브러리 및 기타 지원 파일의 복사본이 포함되어 있다. 가상 환경에 원하는 이름을 사용할 수 있습니다.

이 가상 환경을 사용하려면 활성화 스크립트를 실행하여 활성화해야 합니다.

source venv/bin/activate

 

 

활성화되면 $PATH 변수의 시작 부분에 가상 환경의 bin 디렉토리가 추가됩니다. 또한 셸 프롬프트가 변경되고 현재 사용 중인 가상 환경의 이름이 표시됩니다. 이 경우에 그것은 venv이다.

TensorFlow를 설치하려면 pip 버전 19 이상이 필요합니다. 다음 명령을 실행하여 pip을 최신 버전으로 업그레이드합니다.

pip install --upgrade pip

 

 

 

3. TensorFlow 설치

이제 가상 환경이 활성화되었으므로 TensorFlow 라이브러리를 설치해야 합니다. 이렇게 하려면 다음을 입력합니다.

pip install --upgrade tensorflow

 

 

전용 NVIDIA GPU가 있고 처리 능력을 활용하려면 텐서플로 대신 GPU 지원을 포함하는 텐서플로-gpu 패키지를 설치하십시오.

가상 환경에서 pip3 대신 명령 pip을 사용할 수 있고 python3 대신 python을 사용할 수 있습니다.

설치를 확인하려면 TensorFlow 버전을 인쇄하는 다음 명령을 사용합니다.

작성할 당시, 텐서플로의 최신 안정 버전은 2.0.0이다.

python -c 'import tensorflow as tf; print(tf.__version__)'

# 2.0.0

 

 

TensorFlow 버전은 여기에 표시된 버전과 다를 수 있습니다.

TensorFlow를 처음 사용하는 경우 Get Started with TensorFlow 페이지를 방문하여 첫 번째 ML 애플리케이션을 작성하는 방법에 대해 알아봅니다. 또한 Github에서 TensorFlow Model 또는 TensorFlow-Presample 저장소를 복제하고 TensorFlow 예제를 탐색하고 테스트할 수 있다.

작업이 완료되면 비활성화를 입력하여 환경을 비활성화하면 정상 셸로 돌아갑니다.

deactivate

 

반응형