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

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

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

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

 

 

---

 

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

성공!

---