엔지니어스 - Engineeus

[Ubuntu] NVIDIA Jetson TX2 세팅 [3] - TensorFlow 설치 본문

NVIDIA/TX2

[Ubuntu] NVIDIA Jetson TX2 세팅 [3] - TensorFlow 설치

Engineeus 2020. 3. 27. 23:42
728x90

이전에 JetPack설치를 다뤘습니다. 근데 TensorFlow는 같이 안깔리길래 따로 깔아 주도록 했습니다.

https://mickael-k.tistory.com/87

 

[Ubuntu] NVIDIA Jetson TX2 세팅 [2] - JetPack 설치

start() { 이전 포스트에서 기본 설치 과정을 다뤘습니다. https://mickael-k.tistory.com/86 [Ubuntu] NVIDIA Jetson TX2 세팅 start() { 본 포스트에서는 NVIDIA사의 Jetson TX2 개발자 키트를 이용하여 초기 세..

mickael-k.tistory.com

 

 


 

1. 필요한 package 설치

$ sudo apt update // system package 

$ sudo apt install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev // install and upgrade pip3 
$ sudo apt install python3-pip 

$ sudo pip3 install -U pip testresources setuptools //install python package dependencies 
$ sudo pip3 install -U numpy==1.16.1 future==0.17.1 mock==3.0.5 h5py==2.9.0 keras_preprocessing==1.0.5 keras_applications==1.0.8 gast==0.2.2 enum34 futures protobuf

 

2. Python change

기존 Ubuntu는 default로 Python 2.x를 채택합니다. 따라서 3.x로 바꿔 주도록 합니다.

3. TensorFlow 설치

jetpack 4.2.x 버전이면 아래 코드에서 v42, jetpack 4.3.x 버전이면 v43, 3.3.x 버전이면 v33 이렇게 바꿔 주어야 합니다. 

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.14.0+nv19.10

그리고 그 뒤에 tensorflow-gpu=='1.14.0' 이게 텐플 버전이고 그 뒤에 '19.10'이 엔비디아 버전 입니다. 아래를 참고하여 맞춰주었습니다.

 

Nvidia 사이트를 보면 텐서플로우 2도 이제 설치 할 수 있다고 하는데 아무래도 TenforRT나 기타 라이브러리와의 호환성 문제로 1.x버전으로 깔아야 될듯 합니다. 

 

 

확인

python
import tensorflow as tf

성공!


 

'NVIDIA > TX2' 카테고리의 다른 글

[UBUNTU] NVIDIA Jetson TX2 [6] - GPU Usage 설치  (0) 2020.03.28
[Ubuntu] NVIDIA Jetson TX2 세팅 [1]  (0) 2020.03.21
0 Comments
댓글쓰기 폼