Dockerfile

From nvcr.io/nvidia/pytorch:20.09-py3

Expose 6006 6007 6008 6009

Run apt-get update && apt-get install -y \
    software-properties-common
Run add-apt-repository universe
Run apt-get update && apt-get install -y \
    curl \
    git \
    ffmpeg \
    libjpeg-dev \
    libpng-dev 

Run pip3 install --upgrade pip
Run pip3 uninstall tensorboard -y
Run pip3 uninstall tensorboard-plugin-dlprof -y
Run pip3 install ffmpeg
Run pip3 install prefetch_generator
Run pip3 install librosa==0.8.0
Run pip3 install omegaconf==2.0.6
Run pip3 install pytorch_lightning==1.2.10
Run pip3 install tgt==1.4.4
Run pip3 install pypinyin==0.39.0
Run pip3 install g2p_en==2.1.0
Run pip3 install Unidecode==1.1.1

Run ldconfig && \
apt-get clean && \
apt-get autoremove && \
rm -rf /var/lib/apt/lists/* /tmp/*

WORKDIR /workspace

COPY *py /workspace/
COPY *yaml /workspace/
COPY utils /workspace/utils

ARG USER_ID
ARG GROUP_ID
RUN addgroup --gid $GROUP_ID your_id
RUN adduser --disabled-password --gecos '' --uid $USER_ID --gid $GROUP_ID your_id

USER your_id