diff --git a/docker-compose.yaml b/docker-compose.yaml
new file mode 100644
index 000000000..b9a7599d1
--- /dev/null
+++ b/docker-compose.yaml
@@ -0,0 +1,10 @@
+version: '3'
+services:
+  train:
+    image: tensorflow/tensorflow
+    volumes:
+    - "$PWD:/work"
+    ports:
+    - "6006:6006"
+    working_dir: /work
+    command: /bin/bash