From 3dc1a944142d9862064dcf165d9fc7fce6608c41 Mon Sep 17 00:00:00 2001
From: Kaszanas <andrzej.bialecki94@gmail.com>
Date: Fri, 20 Oct 2023 22:12:43 +0200
Subject: [PATCH] Added dockerfile, updated README

---
 Dockerfile | 14 ++++++++++++++
 README.md  | 15 +++++++++++++++
 2 files changed, 29 insertions(+)
 create mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..4e9ee1b
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,14 @@
+FROM nvidia/cuda:12.2.2-base-ubuntu22.04
+
+RUN apt-get update
+
+RUN apt-get install -y --no-install-recommends \
+    python3 python3-dev python3-pip git build-essential ffmpeg
+
+RUN pip install ninja allin1 git+https://github.com/CPJKU/madmom
+RUN pip install install natten -f https://shi-labs.com/natten/wheels/cu118/torch2.0.0/index.html
+RUN pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
+
+RUN apt-get clean && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /app
\ No newline at end of file
diff --git a/README.md b/README.md
index 0a353e0..69622af 100644
--- a/README.md
+++ b/README.md
@@ -32,6 +32,21 @@ This package provides models for music structure analysis, predicting:
 
 ## Installation
 
+### Docker
+
+Build the image:
+
+```
+docker build --tag=allin1:latest .
+```
+
+Run the container and mount current directory as a volume:
+```
+docker run -it -v .:/app allin1:latest
+```
+
+Follow steps in [Usage for CLI](#usage-for-cli) or [Usage for Python](#usage-for-python)
+
 ### 1. Install PyTorch
 
 Visit [PyTorch](https://pytorch.org/) and install the appropriate version for your system.