Linux/Ubuntu

Ubuntu 18.04에서 tensorflow 설치하기 (Nvidia Jetson Tx2, wheel)

jinmc 2020. 10. 19. 12:56
반응형

일단 official document for Jetson은 이거다

docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html

 

Installing TensorFlow For Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

Installing TensorFlow for Jetson Platform provides you with the access to the latest version of the framework on a lightweight, mobile platform without being restricted to TensorFlow Lite.

docs.nvidia.com

이대로만 따라하면 별문제 없을듯?

 

amanokaze.github.io/blog/Install-Tensorflow-copy/

 

Ubuntu 18.04에서 Tensorflow 2.0 설치하기(Using WSL)

Various IT-based research space, django, cloud, machine learning, and so on.

amanokaze.github.io

Nvidia Jetson에서는 Default로 주어지는 환경들이 있다. 그 중에서 Ubuntu 18.04는 기본적으로 주어지기 때문에 

(Deepstream, Jetpack이라는 프로그램을 깔기 위해서) Ubuntu 18.04를 자주 사용한다.

 

여기다가 Tensorflow 등을 깔고 프로젝트의 특성에 맞춰서 다른 library들을 깔면 된다. 

물론 venv를 사용하면 좋겠지만, venv를 사용하지 못하는 경우에는 정신 바짝 차리고 하나하나 잘 설치해야 한다.

 

일단 Tensorflow를 설치하기 위해서는 Python, Pip는 당연히 깔아줘야 한다.

Python은 Jetson Tx2같은 경우 3.6을 많이 깔고, 나도 왠지는 모르지만 3.7, 3.8은 에러가 많아 3.6을 많이 쓴다고 들었다.

PHP 쓸 때도 7.3, 7.4는 아직 안정화가 되지 않아서 7.2를 많이 썼는데, 딱 그 느낌이다.

 

$ pip install --upgrade pip
$ pip install setuptools --upgrade
$ pip install wheel
$ pip install "numpy<1.17"

numpy같은 경우 1.17이상의 경우 문법 차이에 의한 경고 메세지가 나오기 때문에 저렇게 해준다고 함.

특히, 저 wheel과 setuptools library가 다른 library들을 까는데도 엄청 도움이 되는것 같다.

 

Shapely와 matplotlib라는 library도 깔았는데, 

그 경우에도 wheel이 도움이 된다.

또한, apt get을 사용하여 shapely나 matplotlib를 다운받고 그 다음 pip install을 사용하면 되는 경우도 있다.

 

여튼 그 이후를 한다.

pip install tensorflow

 

근데 이럴 경우 최신 버전 (현재 2020.10.19 기준 2.3.0) 이 깔리기 때문에

Tensorflow 1을 사용하고 싶은 경우 지정해줘야 한다.

 

forums.developer.nvidia.com/t/tensorflow-for-jetson-tx2/64596 

 

TensorFlow for Jetson TX2!

Our official release of TensorFlow for Jetson TX2! Python 3.6+JetPack4.4 sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran sudo apt-get install python3-pip sudo pip3 install -U pip

forums.developer.nvidia.com

Jetson Tx2의 경우 tf 1.15.2 를 사용한다고 하는데, 그러는 경우

 

pip install tensorflow==1.15.2

pip install tensorflow==1.15.2

 

forums.developer.nvidia.com/t/tensorflow-for-jetson-tx2/64596

 

 

반응형

'Linux > Ubuntu' 카테고리의 다른 글

SFTP 사용법과 사용하는 이유  (0) 2020.12.18
linux log 생성하기  (0) 2020.12.08
Ubuntu cli mode로 시작하기  (0) 2020.11.23
Ubuntu 메모리, 저장공간 확인  (0) 2020.10.22
Ubuntu 18.04에서 python dual로 세팅하기  (0) 2020.10.19