Dockerfile.gpu

ARG BASEIMAGE=nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04

FROM ${BASEIMAGE}

ARG DEPSLIST=requirements-gpu.txt

ENV PYTHONUNBUFFERED 1

COPY *.pbmm ./
COPY *.scorer ./
COPY setup.py ./

RUN DEBIAN_FRONTEND=noninteractive apt update && \
    apt -y install ffmpeg libsm6 libxext6 python3 python3-pip wget && \
    apt -y clean && \
	rm -rf /var/lib/apt/lists/*

COPY $DEPSLIST ./requirements.txt
COPY run.sh ./run.sh
RUN chmod +x ./run.sh

# make sure pip is up-to-date
RUN python3 -m pip install --upgrade pip
RUN pip3 install --no-cache-dir -r requirements.txt

# Mount data
VOLUME /input
VOLUME /output
VOLUME /deepspeech
RUN mkdir /audio

# Copying autosub for last for faster debugging
COPY autosub ./autosub

ENTRYPOINT ["./run.sh"]