| 1 | FROM pytorch/pytorch:2.4.0-cuda12.4-cudnn9-devel |
| 2 | |
| 3 | USER root |
| 4 | |
| 5 | ARG DEBIAN_FRONTEND=noninteractive |
| 6 | |
| 7 | LABEL github_repo="https://github.com/SWivid/F5-TTS" |
| 8 | |
| 9 | RUN set -x \ |
| 10 | && apt-get update \ |
| 11 | && apt-get -y install wget curl man git less openssl libssl-dev unzip unar build-essential aria2 tmux vim \ |
| 12 | && apt-get install -y openssh-server sox libsox-fmt-all libsox-fmt-mp3 libsndfile1-dev ffmpeg \ |
| 13 | && apt-get install -y librdmacm1 libibumad3 librdmacm-dev libibverbs1 libibverbs-dev ibverbs-utils ibverbs-providers \ |
| 14 | && rm -rf /var/lib/apt/lists/* \ |
| 15 | && apt-get clean |
| 16 | |
| 17 | WORKDIR /workspace |
| 18 | |
| 19 | RUN git clone https://github.com/SWivid/F5-TTS.git \ |
| 20 | && cd F5-TTS \ |
| 21 | && git submodule update --init --recursive \ |
| 22 | && pip install -e . --no-cache-dir |
| 23 | |
| 24 | ENV SHELL=/bin/bash |
| 25 | |
| 26 | VOLUME /root/.cache/huggingface/hub/ |
| 27 | |
| 28 | EXPOSE 7860 |
| 29 | |
| 30 | WORKDIR /workspace/F5-TTS |
| 31 |