diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..66cf21f
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,18 @@
+FROM pytorch/pytorch:1.7.0-cuda11.0-cudnn8-runtime
+
+# Update system
+RUN apt update
+RUN apt install -y build-essential
+
+# Install Requirements
+COPY requirements.txt .
+RUN pip install --ignore-installed -r requirements.txt
+RUN pip install transformers
+RUN pip install --upgrade numpy
+
+# Copy over files and run
+COPY . .
+RUN python ./main_run.py --help
+
+# Enter and run what you like
+CMD /bin/bash
diff --git a/README.md b/README.md
index f83ebf0..96e5cbc 100644
--- a/README.md
+++ b/README.md
@@ -37,6 +37,21 @@ python main_run.py --folder=bgl/ --log_file=BGL.log --dataset_name=bgl --model_n
 python main_run.py --help
 ```
 
+
+### Docker
+
+#### Build
+
+```
+docker build -t lad .
+```
+
+#### Run
+
+```
+docker run --gpus all -it lad
+```
+
 ## Citation
 If you find the code and models useful for your research, please cite the following paper:
 ```