From 93152d9a9c36e7ac66790dac6f61ec9242b07001 Mon Sep 17 00:00:00 2001 From: Fedir Zadniprovskyi Date: Sat, 18 May 2024 15:21:39 -0700 Subject: [PATCH] Docker file improvements Build time reduced from 165s to 53s (on g6.4xlarge EC2 instance) Final image size reduced from 6.07GB to 4.6GB --- 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"]