Update for Jetson Orin Nano

Addedd an update for make this script work on Orin Nano and nvcr.io/nvidia/l4t-tensorrt:r8.5.2.2-devel container
This commit is contained in:
Bertrand d'Hérouville 2023-04-30 11:45:22 +02:00 committed by GitHub
parent cbd07696b5
commit ec8d062a1c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -13,6 +13,15 @@ mkdir -p ${OUTPUT_FOLDER}
# Install packages # Install packages
pip install --upgrade pip && pip install onnx==1.9.0 protobuf==3.20.3 pip install --upgrade pip && pip install onnx==1.9.0 protobuf==3.20.3
#test if arch is aarch64 (tested on Jetson Orin Nano and nvcr.io/nvidia/l4t-tensorrt:r8.5.2.2-devel )
if [[ $(uname -m) == "aarch64" ]]; then
echo "This system has a 64-bit ARM processor."
pip3 install -U onnx
apt-get update && apt-get install -y wget
else
echo "This system does not have a 64-bit ARM processor."
fi
# Clone tensorrt_demos repo # Clone tensorrt_demos repo
git clone --depth 1 https://github.com/yeahme49/tensorrt_demos.git /tensorrt_demos git clone --depth 1 https://github.com/yeahme49/tensorrt_demos.git /tensorrt_demos