Tensorflow Installation on Jetson TX2

Update

2019.8.20: 작성

2019.8.29: 링크 및 오류 수정

2019.9.1: 포스팅

 

  Jetson TX2에 ubuntu와 JetPack을 깔았다면 이제 임베디드 시스템에서 딥러닝을 할 준비가 어느정도 된 것이다. 이제부터는 본인이 어떤 프레임워크로 딥러닝을 구동할지 결정하여 프레임워크를 설치하면 된다. 본인은 Tensorflow를 기반으로 딥러닝 알고리즘을 구현하고 있기 때문에 본 포스팅에서는 JetsonTX2에 Tensorflow를 설치하는 방법에 대해 포스팅 하려고 한다. 참고로 JetsonTX2에 ubuntu와 JetPack을 설치하는 방법은 유투브 아래의 링크를 참조하길 바란다. 본인은 ubuntu16.04에 Jetpack 3.3을 설치하였다.

 

https://www.youtube.com/watch?v=D7lkth34rgM&t=69s

 

본 포스팅은 다음의 github를 따라가며 진행하였고, 중간에 발생하는 에러등을 첨부하여 작성하였다.

 

https://github.com/jetsonhacks/installTensorFlowTX2

 

본인은 Jetson TX2에 tensorflow를 설치하는 것만 관심이 있기 때문에 다음의 링크에서 .whl파일을 다운받아서 tensorflow를 설치하였다. 다음의 링크는 nvidia의 Jetson Download Center이다. Jetson Download Center의 왼쪽 search에 tensorflow를 검색하고 'Tensorflow for Jetpack'에서 JP3.3 버전을 선택하여 아래의 파일이름을 가진 .whl 파일을 다운로드 한다.

 

https://developer.nvidia.com/embedded/downloads#?search=tensorflow

파일이름: tensorflow_gpu-1.9.0+nv18.8-cp35-cp35m-linux_aarch64.whl

 

파이썬을 설치하기 앞에 우선 다음 명령어를 입력한다.

 

sudo apt-get update

 

 

update가 끝나면 다음 명령어를 입력하여 python3를 설치한다.

 

sudo apt-get install -y python3-pip python3-dev

 

다음 명령어로 다운받은 .whl이 위치한 폴더로 이동하여 tensorflow를 다음 명령어를 이용하여 설치한다.

 

python3 -m pip install tensorflow-wheel-file --user

예를 들면, python3 -m pip install tensorflow_gpu-1.90.+nv18.8-cp35-cp35m-linux_aarch64.whl --user

(작성한 명령어에는 오타가 있을 수 있으니 잘 살펴보길 바란다.)

 

 

설치가 잘 되면 좋은데, 설치를 하다보면 다음과 같은 에러가 발생한다.

 

 

위의 에러를 해결하기 위해서는 enum34, futures, 그리고 setuptools를 설치해야 한다. 다음의 명령어로 설치해준다.

 

sudo apt-get install python-enum34

sudo apt-get install python-futures

sudo -H pip3 install -U pip setuptools

 

모두 설치가 끝나면 tensorflow를 다시 설치한다. 설치가 완료되면 tensorflow불러와져야 한다. tensorflow가 설치 되었는 확인하기 위해 다음과 같이 간단한 코딩을 해본다. 먼저 터미널에 python3 명령어를 입력하여 파이썬을 실행한다.

 

import tensorflow as tf

a = tf.constant(1)
b = tf.constant(2)
sum = a + b

sess = tf.Session()
print(sess.run(sum))

### 결과
### >>>3

 

 

위와 같이 예제가 잘 구현되었다면 tensorflow가 잘 설치된 것이다.

 

 

 

참조

[1] https://www.youtube.com/watch?v=D7lkth34rgM&t=69s

[2] https://github.com/jetsonhacks/installTensorFlowTX2

[3] https://github.com/jetsonhacks/installTensorFlowJetsonTX/tree/master/TX2