diff --git a/docker/Dockerfile b/docker/Dockerfile index b4d6aa7..604c8e1 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"]