From 779baff9c357bf8335e18c3f079bffb00f2f5828 Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Wed, 22 Jan 2025 05:11:26 -0500 Subject: [PATCH] Add pynvml missing dep for tensorrt Signed-off-by: makaveli10 --- docker/Dockerfile.tensorrt | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/Dockerfile.tensorrt b/docker/Dockerfile.tensorrt index 3b9b125..c46a2d2 100644 --- a/docker/Dockerfile.tensorrt +++ b/docker/Dockerfile.tensorrt @@ -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 . \ No newline at end of file