Add pynvml missing dep for tensorrt

Signed-off-by: makaveli10 <vineet.suryan@collabora.com>
2025-01-22 05:11:26 -05:00
parent 5aa5826f36
commit 779baff9c3
1 changed files with 1 additions and 0 deletions
@@ -25,6 +25,7 @@ RUN apt update && bash setup.sh && rm setup.sh

 COPY requirements/server.txt .
 RUN pip install --no-cache-dir -r server.txt && rm server.txt
+RUN pip install pynvml==11.5.0
 COPY whisper_live ./whisper_live
 COPY scripts/build_whisper_tensorrt.sh .
 COPY run_server.py .