Update onnxruntime gpu and mediapipe (#439)

* ensure onnxruntime-gpu version matches onnx * update mediapipe
livepeer · Mar 4, 2025 · 328c547 · 328c547
1 parent 86b5821
commit 328c547
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/runner/docker/Dockerfile.live-app__PIPELINE__ b/runner/docker/Dockerfile.live-app__PIPELINE__
@@ -11,10 +11,13 @@ RUN apt-get update && apt-get install -y \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
 # Install any additional Python packages
+
 COPY requirements.live-ai.txt /app/requirements.txt
 RUN pip install --no-cache-dir -r /app/requirements.txt
 # TODO: Figure out a way to have this in requirements file
 RUN pip install --no-cache-dir triton==3.1.0
+RUN pip uninstall -y onnx onnxruntime onnxruntime-gpu
+RUN pip install onnx==1.17.0 onnxruntime-gpu==1.17.0
 
 # Set environment variables
 ENV MAX_WORKERS=1

diff --git a/runner/requirements.live-ai.txt b/runner/requirements.live-ai.txt
@@ -26,3 +26,4 @@ aiohttp==3.10.9
 nvidia-ml-py==12.560.30
 pynvml==12.0.0
 prometheus_client>=0.21.1
+mediapipe==0.10.20