vixy/dreamtail
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix onnxruntime for Jetson ARM64

Browse Source
This commit is contained in:
Vixy
2026-01-01 20:13:02 -06:00
parent 3b587a2d51
commit b612127a80
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@@ -27,6 +27,10 @@ COPY requirements.txt /app/
# Note: torch and torchvision are already in the base image # Note: torch and torchvision are already in the base image
RUN pip3 install --no-cache-dir -r requirements.txt 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 application code
COPY config.py /app/ COPY config.py /app/
COPY main.py /app/ COPY main.py /app/

2
requirements.txt
View File

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