From 37378898a204c91e4aa47f163dd635a06c437628 Mon Sep 17 00:00:00 2001
From: thomwolf <thomwolf@gmail.com>
Date: Fri, 14 Dec 2018 15:02:32 +0100
Subject: [PATCH] adding DockerFile

---
 docker/Dockerfile | 7 +++++++
 1 file changed, 7 insertions(+)
 create mode 100644 docker/Dockerfile

diff --git a/docker/Dockerfile b/docker/Dockerfile
new file mode 100644
index 0000000000..e47eb548f9
--- /dev/null
+++ b/docker/Dockerfile
@@ -0,0 +1,7 @@
+FROM pytorch/pytorch:latest
+
+RUN git clone https://github.com/NVIDIA/apex.git && cd apex && python setup.py install --cuda_ext --cpp_ext
+
+RUN pip install pytorch-pretrained-bert
+
+WORKDIR /workspace
\ No newline at end of file