Create Dockerfile

This commit is contained in:
Zhikang Niu
2024-10-18 11:23:01 +08:00
committed by GitHub
parent 18f526dd25
commit 5fa56825be

25
Dockerfile Normal file
View File

@@ -0,0 +1,25 @@
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