diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..4a407e9 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,32 @@ +FROM nvidia/cuda:11.3.1-base-ubuntu20.04 + +ENV PYTHON_VERSION=3.8 + +RUN export DEBIAN_FRONTEND=noninteractive \ + && apt-get -qq update \ + && apt-get -qq install --no-install-recommends \ + git \ + python${PYTHON_VERSION} \ + python${PYTHON_VERSION}-venv \ + python3-pip \ + && rm -rf /var/lib/apt/lists/* + +RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \ + ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python && \ + ln -s -f /usr/bin/pip3 /usr/bin/pip + +RUN pip install --upgrade pip + +# 2. Copy files +COPY . /src + +RUN pip install torch==1.12.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 + +WORKDIR /src +# 3. Install dependencies +RUN pip install -r requirements-docker.txt + +RUN python3 setup.py install + + + diff --git a/requirements-docker.txt b/requirements-docker.txt new file mode 100644 index 0000000..c6bb9d2 --- /dev/null +++ b/requirements-docker.txt @@ -0,0 +1,15 @@ +tqdm +rotary_embedding_torch +transformers==4.19 +tokenizers +inflect +progressbar +einops +unidecode +scipy +librosa +numpy==1.20.0 +numba==0.48.0 +threadpoolctl +llvmlite +appdirs