From 137530331f3d8d43140a722aad3d169a95551474 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicolas=20Mass=C3=A9?= <nicolas.masse@itix.fr>
Date: Thu, 3 Jul 2025 16:57:56 +0200
Subject: [PATCH] fix the app to work on Nvidia Jetson Orin

---
 app/Containerfile                      | 27 ++++++++++++++++----------
 app/app.py                             |  1 +
 app/requirements.txt                   |  4 ++--
 base-image/{ => jetpack}/Containerfile |  0
 base-image/{ => jetpack}/build.sh      |  0
 5 files changed, 20 insertions(+), 12 deletions(-)
 rename base-image/{ => jetpack}/Containerfile (100%)
 rename base-image/{ => jetpack}/build.sh (100%)

diff --git a/app/Containerfile b/app/Containerfile
index 0679c9b..345f577 100644
--- a/app/Containerfile
+++ b/app/Containerfile
@@ -1,35 +1,42 @@
-FROM quay.io/nmasse-redhat/jetpack-multiarch:r36.4.0
+FROM nvcr.io/nvidia/l4t-jetpack:r36.3.0
 
+ARG PYPI_MIRROR_URL
 ADD *.py *.jpg *.onnx requirements.txt /opt/app
 WORKDIR /opt/app
 RUN <<EOF
 set -Eeu
 
-# Tell dpkg-configure to avoid asking questions...
+# Install pip
+echo "Installing pip for $(python3 -V)..."
 export DEBIAN_FRONTEND=noninteractive
+apt-get update
+apt-get install -y python3-pip
+rm -rf /var/lib/apt/lists/*
+apt-get clean
 
 # No caching
 export PIP_NO_CACHE_DIR=1
 
-# Install git + pip
-apt update
-apt install -y git python3-pip
-rm -rf /var/lib/apt/lists/*
-apt-get clean
+# Local PyPi repository ?
+if [ -n "${PYPI_MIRROR_URL:-}" ]; then
+  pip config set global.index-url "${PYPI_MIRROR_URL}"
+fi
 
 # Install Python dependencies
 case "$(arch)" in
 aarch64)
   echo "Downloading onnxruntime_gpu from Nvidia"
-  wget -q https://nvidia.box.com/shared/static/i7n40ki3pl2x57vyn4u7e9asyiqlnl7n.whl -O onnxruntime_gpu-1.16.0-cp310-cp310-linux_aarch64.whl
-  pip3 install onnxruntime_gpu-1.16.0-cp310-cp310-linux_aarch64.whl
-  rm -f onnxruntime_gpu-1.16.0-cp310-cp310-linux_aarch64.whl
+  # See https://elinux.org/Jetson_Zoo#ONNX_Runtime for updates
+  wget -q https://nvidia.box.com/shared/static/6l0u97rj80ifwkk8rqbzj1try89fk26z.whl -O onnxruntime_gpu-1.19.0-cp310-cp310-linux_aarch64.whl
+  pip3 install onnxruntime_gpu-1.19.0-cp310-cp310-linux_aarch64.whl
+  rm -f onnxruntime_gpu-1.19.0-cp310-cp310-linux_aarch64.whl
   ;;
 x86_64)
   pip3 install onnxruntime_gpu
   ;;
 esac
 pip3 install -r requirements.txt
+rm -f $HOME/.config/pip/pip.conf
 EOF
 
 ENTRYPOINT [ "python3", "app.py" ]
diff --git a/app/app.py b/app/app.py
index 6d76644..a1fe518 100644
--- a/app/app.py
+++ b/app/app.py
@@ -1,3 +1,4 @@
+#!/usr/bin/python3
 import numpy as np
 import time
 import onnxruntime as ort
diff --git a/app/requirements.txt b/app/requirements.txt
index a2dc04a..13ca20c 100644
--- a/app/requirements.txt
+++ b/app/requirements.txt
@@ -1,3 +1,3 @@
 opencv-python-headless
-numpy
-onnxruntime_gpu
+numpy<2
+onnxruntime-gpu
diff --git a/base-image/Containerfile b/base-image/jetpack/Containerfile
similarity index 100%
rename from base-image/Containerfile
rename to base-image/jetpack/Containerfile
diff --git a/base-image/build.sh b/base-image/jetpack/build.sh
similarity index 100%
rename from base-image/build.sh
rename to base-image/jetpack/build.sh