| 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 \ | |
| && rm -rf /var/lib/apt/lists/* \ | |
| && apt-get clean | |
| WORKDIR /workspace | |
| RUN git clone https://github.com/SWivid/F5-TTS.git \ | |
| && cd F5-TTS \ | |
| && pip install --no-cache-dir -r requirements.txt \ | |
| && pip install --no-cache-dir -r requirements_eval.txt | |
| ENV SHELL=/bin/bash | |
| WORKDIR /workspace/F5-TTS | |