Jetson 보드에 Tensorflow 설치하기

Ubuntu Version : 20.04

Board : Jetson Orin NX 8gb

[110110190]reComputer?Industrial?J4011- Fanless Edge AI Device with Jetson Orin™ NX 8GB module, Aluminum case with passive coo

Jetpack : 5.1.1

 

딥러닝을 돌리려고 했는데 GPU가 잡히질 않아 삭제 후 nvidia 공식 문서를 참고하여 다시 설치한 후 정상적으로  GPU를 사용하는 것을 확인할 수 있습니다

 

1. Conda install (Option)

anaconda3를 활용하여 pyhton3.8 가상환경을 만들어 준다

conda create -n python3.8 python=3.8

 

 

2. Tensorflow 설치하기

- 관련 패키지 설치

$ sudo apt-get update
$ sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran

 

- pip 을 활용한 파이썬 패키지 설치

$ sudo apt-get install python3-pip
$ sudo python3 -m pip install --upgrade pip
$ pip3 install -U testresources setuptools==65.5.0

 

- 파이썬 종송석을 위한 관련 패지키 설치

$ pip3 install -U numpy==1.22 future==0.18.2 mock==3.0.5 keras_preprocessing==1.1.2 keras_applications==1.0.8 gast==0.4.0 protobuf pybind11 cython pkgconfig packaging h5py==3.7.0

 

그다음 중요

1. 아래 페이지 접속

https://developer.download.nvidia.com/compute/redist/jp

위 페이지에서 자신의 jetpack 확인 jtop을 활용하여 확인할 수 있음 

참고 ) 보통 젯슨나노는 4.6 jetson xavier 나 orin은 5.1.1 또는 5.1.2 사용

jtop 결과

2. v511 클릭 -> tensorflow 클릭

그리면 오른쪽 사진과 같이 텐서플로우 버전이 나온다.

이를 참고 하여 아래 명령어에 적절하게 인덱스를 수정해준다

$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v511 tensorflow==2.12.0+nv23.05

 

 

젯슨나노의 경우 아래 명령어를 해주면 될 듯

$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v46 tensorflow==2.6.2+nv21.12

 

참고 사진

 

 

3. 테스트 하기

방법 1 : 간단

 

방법 2 : gpu_test1.py 만들어서 아래 코드 입력후 python3 gpu_test1.py로 실행

import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("No GPU available")

 

방법3 : gpu_test2.py

import tensorflow as tf

print(tf.config.list_logical_devices())
print(tf.config.list_physical_devices())
print(tf.test.is_built_with_cuda())
print(tf.test.is_built_with_gpu_support())
print(tf.test.is_built_with_rocm())
print(tf.test.is_built_with_xla())
print(tf.version.GIT_VERSION)
print(tf.version.VERSION)

 

방법 3을 보면 tensorflow와 CUDA가 호환되지 않는 것처럼 보인다 이를 위해서 다음 포스트에서 관련 내용을 다루고자 한다.

 

읽어 주셔서 감사합니다 :)

 

<Reference>

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

Installing TensorFlow for Jetson Platform - NVIDIA Docs