Docker file improvements (#848)

Docker file improvements Co-authored-by: Fedir Zadniprovskyi <[email protected]>
SYSTRAN · May 20, 2024 · 65551c0 · 65551c0
1 parent f53be1e
commit 65551c0
Showing 1 changed file with 3 additions and 5 deletions.
diff --git 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"]