dockerfile init

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 AS stage1
 ARG DEBIAN_FRONTEND=noninteractive
 ARG TZ=America/Los_Angeles
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
@@ -18,10 +18,37 @@ RUN conda install python=3.9.13
 RUN python3 -m pip install --upgrade pip
 RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
 
-# RUN git clone https://git.ecker.tech/mrq/ai-voice-cloning.git
-# WORKDIR /home/user/ai-voice-cloning
-# ENV VIRTUAL_ENV=/home/user/ai-voice-cloning/venv
-# ENV PATH="$VIRTUAL_ENV/bin:$PATH"
-# RUN ./setup-cuda.sh
+FROM stage1 AS stage2
+USER root
+RUN mkdir /home/user/ai-voice-cloning
+RUN mkdir /home/user/ai-voice-cloning/modules
+RUN chown -R user:user /home/user/ai-voice-cloning
 
-CMD ["python"]
+USER user
+WORKDIR /home/user/ai-voice-cloning
+
+ADD .gitmodules .gitmodules
+ADD .git .git
+ADD modules modules
+USER root
+RUN chown -R user:user /home/user/ai-voice-cloning
+USER user
+RUN git config --global --add safe.directory /home/user/ai-voice-cloning
+RUN git submodule init
+RUN git submodule update --remote
+# install TorToiSe requirements
+RUN python3 -m pip install -r ./modules/tortoise-tts/requirements.txt
+# install TorToiSe
+RUN python3 -m pip install -e ./modules/tortoise-tts/
+# instal DLAS requirements, last, because whisperx will break a dependency here
+RUN python3 -m pip install -r ./modules/dlas/requirements.txt
+# install DLAS
+RUN python3 -m pip install -e ./modules/dlas/
+
+ADD requirements.txt requirements.txt
+# install local requirements
+RUN python3 -m pip install -r ./requirements.txt
+
+ADD . /home/user/ai-voice-cloning
+
+CMD ["python", "./src/main.py", "--listen", "0.0.0.0:7680"]