Fix onnxruntime for Jetson ARM64
This commit is contained in:
@@ -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/
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user