18 lines
533 B
Docker
18 lines
533 B
Docker
FROM pytorch/pytorch:latest
|
|
|
|
RUN apt-get update && apt-get install -y build-essential ffmpeg parallel aria2 && apt-get clean
|
|
|
|
COPY ./requirements.txt /workspace/requirements.txt
|
|
|
|
RUN pip install -r requirements.txt && pip install webrtcvad-wheels
|
|
|
|
COPY . /workspace
|
|
|
|
VOLUME [ "/datasets", "/workspace/synthesizer/saved_models/" ]
|
|
|
|
ENV DATASET_MIRROR=default FORCE_RETRAIN=false TRAIN_DATASETS=aidatatang_200zh\ magicdata\ aishell3\ data_aishell TRAIN_SKIP_EXISTING=true
|
|
|
|
EXPOSE 8080
|
|
|
|
ENTRYPOINT [ "/workspace/docker-entrypoint.sh" ]
|