Docker file improvements (#848)

Docker file improvements

Co-authored-by: Fedir Zadniprovskyi <github.g1k56@simplelogin.com>
This commit is contained in:
Fedir Zadniprovskyi
2024-05-19 19:13:19 -07:00
committed by GitHub
parent f53be1e811
commit 65551c081f

View File

@@ -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 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 ./ COPY infer.py jfk.flac ./
RUN pip3 install faster-whisper RUN pip3 install faster-whisper
CMD ["python3", "infer.py"]
ENTRYPOINT ["python3", "infer.py"]