From ec8d062a1ca6b28573e9f6ddbdc8c32084de4180 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bertrand=20d=27H=C3=A9rouville?= Date: Sun, 30 Apr 2023 11:45:22 +0200 Subject: [PATCH] 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 --- docker/tensorrt_models.sh | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/docker/tensorrt_models.sh b/docker/tensorrt_models.sh index 957e817d6..ba283c78c 100755 --- a/docker/tensorrt_models.sh +++ b/docker/tensorrt_models.sh @@ -13,6 +13,15 @@ mkdir -p ${OUTPUT_FOLDER} # Install packages 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 git clone --depth 1 https://github.com/yeahme49/tensorrt_demos.git /tensorrt_demos