FROM pytorch/pytorch
RUN apt update
RUN apt install -y git
RUN pip install git+https://github.com/m-bain/whisperX
#RUN python -c 'from whisperx.asr import load_model; load_model("large-v3", "cpu")'
RUN python -c 'from faster_whisper.utils import download_model; download_model("large-v3")'

Run with:

docker run -it --entrypoint /bin/bash -v'/home/joregan/tmpf:/audio' --gpus all whisperx