Docker image

Dockerfile

@@ -0,0 +1,34 @@
+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
+
+RUN apt-get update && apt-get -y install libsndfile1-dev 
+
+# 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
+
+
+

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