F5-TTS/Dockerfile at main · blahmountain/F5-TTS

Repository navigation

FROM pytorch/pytorch:2.4.0-cuda12.4-cudnn9-devel

USER root

ARG DEBIAN_FRONTEND=noninteractive

LABEL github_repo="https://github.com/SWivid/F5-TTS"

RUN set -x \

&& apt-get update \

&& apt-get -y install wget curl man git less openssl libssl-dev unzip unar build-essential aria2 tmux vim \

&& apt-get install -y openssh-server sox libsox-fmt-all libsox-fmt-mp3 libsndfile1-dev ffmpeg \

&& apt-get install -y librdmacm1 libibumad3 librdmacm-dev libibverbs1 libibverbs-dev ibverbs-utils ibverbs-providers \

&& rm -rf /var/lib/apt/lists/* \

&& apt-get clean

WORKDIR /workspace

RUN git clone https://github.com/SWivid/F5-TTS.git \

&& cd F5-TTS \

&& git submodule update --init --recursive \

&& sed -i '7iimport sys\nsys.path.append(os.path.dirname(os.path.abspath(__file__)))' src/third_party/BigVGAN/bigvgan.py \

&& pip install -e . --no-cache-dir

ENV SHELL=/bin/bash

WORKDIR /workspace/F5-TTS