neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-03-19 03:34:40 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

updated working dockerfile

Browse source
This commit is contained in:
supermomo668 2024-08-02 22:16:21 +00:00
parent 86100b6a23
commit 9c822c22db
1 changed files with 31 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
Dockerfile
View file

@ -1,34 +1,47 @@
FROM nvidia/cuda:12.2.0-base-ubuntu22.04
# Copy application code to /app
COPY . /app
# Install necessary packages and clean up
RUN apt-get update && \
apt-get install -y --allow-unauthenticated --no-install-recommends \
apt-get install -y --no-install-recommends \
wget \
git \
&& apt-get autoremove -y \
&& apt-get clean -y \
&& rm -rf /var/lib/apt/lists/*
&& apt-get clean && rm -rf /var/lib/apt/lists/*
ENV HOME "/root"
ENV CONDA_DIR "${HOME}/miniconda"
ENV PATH="$CONDA_DIR/bin":$PATH
# Set environment variables
ENV HOME="/root"
ENV CONDA_DIR="${HOME}/miniconda"
ENV PATH="$CONDA_DIR/bin:$PATH"
ENV CONDA_AUTO_UPDATE_CONDA=false
ENV PIP_DOWNLOAD_CACHE="$HOME/.pip/cache"
ENV TORTOISE_MODELS_DIR="$HOME/tortoise-tts/build/lib/tortoise/models"
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda3.sh \
&& bash /tmp/miniconda3.sh -b -p "${CONDA_DIR}" -f -u \
&& "${CONDA_DIR}/bin/conda" init bash \
&& rm -f /tmp/miniconda3.sh \
&& echo ". '${CONDA_DIR}/etc/profile.d/conda.sh'" >> "${HOME}/.profile"
# Install Miniconda
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda3.sh && \
bash /tmp/miniconda3.sh -b -p "${CONDA_DIR}" && \
rm -f /tmp/miniconda3.sh && \
echo ". ${CONDA_DIR}/etc/profile.d/conda.sh" >> "${HOME}/.bashrc"
# --login option used to source bashrc (thus activating conda env) at every RUN statement
SHELL ["/bin/bash", "--login", "-c"]
RUN conda create --name tortoise python=3.9 numba inflect \
&& conda activate tortoise \
&& conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia \
&& conda install transformers=4.31.0 \
&& cd /app \
&& python setup.py install
# Initialize conda for the shell session
RUN conda init bash
# Create the conda environment and install required packages
RUN conda create --name tortoise python=3.9 numba inflect -y && \
bash -c "source ${CONDA_DIR}/etc/profile.d/conda.sh && conda activate tortoise && \
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia -y && \
conda install transformers=4.31.0 scipy -y"
# Set conda environment to be activated by default in future RUN instructions
RUN echo "conda activate tortoise" >> ~/.bashrc
# Install the application
WORKDIR /app
RUN bash -c "source ${CONDA_DIR}/etc/profile.d/conda.sh && conda activate tortoise && python setup.py install"
# Default command
CMD ["/bin/bash"]