Fix onnxruntime for Jetson ARM64

This commit is contained in:
2026-01-01 20:13:02 -06:00
parent 3b587a2d51
commit b612127a80
2 changed files with 5 additions and 1 deletions

View File

@@ -27,6 +27,10 @@ COPY requirements.txt /app/
# Note: torch and torchvision are already in the base image
RUN pip3 install --no-cache-dir -r requirements.txt
# Install onnxruntime for Jetson (from NVIDIA's pre-built wheels)
# The base dustynv image should have it, but ensure it's available
RUN pip3 install --no-cache-dir onnxruntime || echo "onnxruntime already installed or using system version"
# Copy application code
COPY config.py /app/
COPY main.py /app/

View File

@@ -23,7 +23,7 @@ opencv-python==4.9.0.80
# IP-Adapter FaceID (for consistent face generation)
insightface==0.7.3
onnxruntime-gpu==1.17.0
# onnxruntime-gpu installed separately for Jetson (from NVIDIA wheels)
# Utilities
pydantic==2.6.0