psr
/
ai-voice-cloning
forked from mrq/ai-voice-cloning
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: psr:d64cba667f5c0ce1d69be07171bc1aec90730960
Branches Tags
psr:docker
psr:master
mrq:master
...
compare: psr:9afafc69c19233fe5501e52d69134e9dc9ef3a2b
Branches Tags
psr:docker
psr:master
mrq:master

2 Commits
d64cba667f ... 9afafc69c1

Author SHA1 Message Date
psr 9afafc69c1 docker: add training script 2023-04-07 23:15:13 +07:00
psr c018bfca9c docker: add ffmpeg for whisper and general cleanup 2023-04-07 23:14:05 +07:00
2 changed files with 29 additions and 12 deletions
Show Stats Expand all files Collapse all files

26
Dockerfile
Unescape Escape View File

@ -1,35 +1,37 @@
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 AS stage1
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
ARG DEBIAN_FRONTEND=noninteractive
ARG TZ=America/Los_Angeles
ARG TZ=UTC
ARG MINICONDA_VERSION=23.1.0-1
ARG PYTHON_VERSION=3.9.13
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
RUN apt-get update
RUN apt install -y curl wget git
RUN apt install -y curl wget git ffmpeg
RUN adduser --disabled-password --gecos '' --shell /bin/bash user
USER user
ENV HOME=/home/user
WORKDIR $HOME
RUN mkdir $HOME/.cache $HOME/.config && chmod -R 777 $HOME
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py39_23.1.0-1-Linux-x86_64.sh
RUN chmod +x Miniconda3-py39_23.1.0-1-Linux-x86_64.sh
RUN ./Miniconda3-py39_23.1.0-1-Linux-x86_64.sh -b -p /home/user/miniconda
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py39_$MINICONDA_VERSION-Linux-x86_64.sh
RUN chmod +x Miniconda3-py39_$MINICONDA_VERSION-Linux-x86_64.sh
RUN ./Miniconda3-py39_$MINICONDA_VERSION-Linux-x86_64.sh -b -p /home/user/miniconda
ENV PATH="$HOME/miniconda/bin:$PATH"
RUN conda init
RUN conda install python=3.9.13
RUN conda install python=$PYTHON_VERSION
RUN python3 -m pip install --upgrade pip
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
FROM stage1 AS stage2
RUN mkdir /home/user/ai-voice-cloning
WORKDIR /home/user/ai-voice-cloning
RUN mkdir $HOME/ai-voice-cloning
WORKDIR $HOME/ai-voice-cloning
COPY --chown=user:user modules modules
FROM stage2 AS stage3
RUN python3 -m pip install -r ./modules/tortoise-tts/requirements.txt
RUN python3 -m pip install -e ./modules/tortoise-tts/
RUN python3 -m pip install -r ./modules/dlas/requirements.txt
RUN python3 -m pip install -e ./modules/dlas/
ADD requirements.txt requirements.txt
RUN python3 -m pip install -r ./requirements.txt
ADD --chown=user:user . /home/user/ai-voice-cloning
ADD --chown=user:user . $HOME/ai-voice-cloning
CMD ["python", "./src/main.py", "--listen", "0.0.0.0:7680"]

15
train-docker.sh
Unescape Escape View File

@ -0,0 +1,15 @@
#!/bin/bash
CMD="python3 ./src/train.py --yaml $1"
# ipc host is one way to increase the shared memory for the container
# more info here https://github.com/pytorch/pytorch#docker-image
CPATH="/home/user/ai-voice-cloning"
docker run --rm --gpus all \
--mount "type=bind,src=$PWD/models,dst=$CPATH/models" \
--mount "type=bind,src=$PWD/training,dst=$CPATH/training" \
--mount "type=bind,src=$PWD/voices,dst=$CPATH/voices" \
--mount "type=bind,src=$PWD/bin,dst=$CPATH/bin" \
--mount "type=bind,src=$PWD/src,dst=$CPATH/src" \
--workdir $CPATH \
--ipc host \
--user "$(id -u):$(id -g)" \
-it ai-voice-cloning $CMD