[ActionAI]Jetson nano에서 ActionAI 실행을 위한 환경설정[tensorrt docker편]

ActionAI 저장소를 clone합니다.

git clone https://github.com/smellslikeml/ActionAI

먼저 docker image를 받습니다.

sudo docker pull nvcr.io/nvidia/l4t-ml:r32.5.0-py3

도커를 실행합니다.

sudo docker run --gpus all -it nvcr.io/nvidia/l4t-ml:r32.5.0-py3

먼저 pip3를 업그레이드 해줍니다.

pip3 install --upgrade pip

모듈들의 버전을 확인합니다.

python3

import tensorflow as tf
tf.__version__
import torch
torch.__version__
import numpy as np
np.__version__
import sklearn
sklearn.__version__

만약 numpy 버전이 1.13대라고 나오면

apt-get remove python3-numpy

tensorflow 2.3.1버전을 설치합니다.

# tensorflow 1.x버전을 제거합니다.
pip3 uninstall tensorflow

apt-get update
apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
apt-get install python3-pip
pip3 install -U pip testresources setuptools==49.6.0 
pip3 install -U numpy==1.18.5 future==0.18.2 mock==3.0.5 h5py==2.10.0 keras_preprocessing==1.1.1 keras_applications==1.0.8 gast==0.2.2 futures protobuf pybind11

# tensorflow 2.3.1+nv20.12버전을 설치합니다.
pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v45 tensorflow==2.3.1+nv20.12

import torch를 할 때

OSError: /usr/lib/aarch64-linux-gnu/libgomp.so.1: cannot allocate memory in static TLS block라는 에러가 뜨면 아래 커맨드를 작성하세요.

export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgomp.so.1

scikit-learn 0.24버전에서 linear_assignment_를 지원하지 않습니다. 0.22.1버전을 설치합니다.

pip3 uninstall scikit-learn
pip3 install scikit-learn==0.22.1

container를 commit합니다.

sudo docker ps 
sudo docker commit {container ID} actionai:latest