From 65551c081f17f30e9e73debbbdd148233de67ac7 Mon Sep 17 00:00:00 2001 From: Fedir Zadniprovskyi <76551385+fedirz@users.noreply.github.com> Date: Sun, 19 May 2024 19:13:19 -0700 Subject: [PATCH] Docker file improvements (#848) Docker file improvements Co-authored-by: Fedir Zadniprovskyi --- docker/Dockerfile | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index b4d6aa7e..604c8e17 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,8 +1,6 @@ -FROM nvidia/cuda:12.0.0-runtime-ubuntu20.04 - +FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 WORKDIR /root -RUN apt-get update -y && apt-get install -y python3 python3-pip libcudnn8 libcudnn8-dev +RUN apt-get update -y && apt-get install -y python3-pip COPY infer.py jfk.flac ./ RUN pip3 install faster-whisper - -ENTRYPOINT ["python3", "infer.py"] +CMD ["python3", "infer.py"]