From b69efa7dbf890a2df792c64c7e28679608e99e47 Mon Sep 17 00:00:00 2001
From: Aymeric Weinbach <aymeric@zecloud.fr>
Date: Thu, 26 Jan 2023 15:10:00 +0000
Subject: [PATCH 1/2] Docker file

---
 Dockerfile              | 32 ++++++++++++++++++++++++++++++++
 requirements-docker.txt | 15 +++++++++++++++
 2 files changed, 47 insertions(+)
 create mode 100644 Dockerfile
 create mode 100644 requirements-docker.txt

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

From dde669abc87ee9d4d19486e3ff4faa326938559e Mon Sep 17 00:00:00 2001
From: Aymeric Weinbach <aymeric@zecloud.fr>
Date: Thu, 26 Jan 2023 17:04:42 +0000
Subject: [PATCH 2/2] missig libsndfile

---
 Dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 4a407e9..12adb65 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,6 +17,8 @@ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \
 
 RUN pip install --upgrade pip
 
+RUN apt-get update && apt-get -y install libsndfile1-dev 
+
 # 2. Copy files
 COPY . /src