diff --git a/Dockerfile b/Dockerfile index ea8485c..0112222 100755 --- a/Dockerfile +++ b/Dockerfile @@ -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/ diff --git a/requirements.txt b/requirements.txt index 86db735..b20c555 100755 --- a/requirements.txt +++ b/requirements.txt @@ -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