From 7bdff38c6edc3e354cd05e0e2aeaa12bcda4cc90 Mon Sep 17 00:00:00 2001
From: leonwanghui <wanghui71leon@gmail.com>
Date: Fri, 26 Mar 2021 10:43:55 +0800
Subject: [PATCH] Add TinyMS docker build script

---
 docker/README.md                       | 14 ++++++++++++++
 docker/tinyms/0.1.0-jupyter/Dockerfile | 11 +++++++++++
 docker/tinyms/0.1.0/Dockerfile         | 10 ++++++++++
 3 files changed, 35 insertions(+)
 create mode 100644 docker/README.md
 create mode 100644 docker/tinyms/0.1.0-jupyter/Dockerfile
 create mode 100644 docker/tinyms/0.1.0/Dockerfile

diff --git a/docker/README.md b/docker/README.md
new file mode 100644
index 00000000..aaa9668e
--- /dev/null
+++ b/docker/README.md
@@ -0,0 +1,14 @@
+## TinyMS Dockerfile Repository
+
+This folder hosts all the `Dockerfile` to build TinyMS container images with various versions and corresponding hardware backends.
+
+> **NOTICE:** Currently TinyMS only supports `CPU` hardware backend.
+
+### TinyMS docker build command
+
+| Hardware Platform | Docker Image Tag | Build Command |
+| :---------------- | :------ | :------------ |
+| CPU | `x.y.z` | cd tinyms/x.y.z && docker build . -t tinyms/tinyms:x.y.z |
+|  | `x.y.z-jupyter` | cd tinyms/x.y.z-jupyter && docker build . -t tinyms/tinyms:x.y.z-jupyter |
+
+> **NOTICE:** The `x.y.z` version shown above should be replaced with the real version number.
diff --git a/docker/tinyms/0.1.0-jupyter/Dockerfile b/docker/tinyms/0.1.0-jupyter/Dockerfile
new file mode 100644
index 00000000..3b36da34
--- /dev/null
+++ b/docker/tinyms/0.1.0-jupyter/Dockerfile
@@ -0,0 +1,11 @@
+ARG BASE_CONTAINER=jupyter/scipy-notebook:ubuntu-18.04
+FROM $BASE_CONTAINER
+
+LABEL MAINTAINER="Leon Wang <wanghui71leon@gmail.com>"
+
+# Set the default jupyter token with "tinyms"
+RUN sh -c '/bin/echo -e "tinyms\ntinyms\n" | jupyter notebook password'
+
+# Install TinyMS cpu whl package
+RUN pip install --no-cache-dir numpy==1.17.5 tinyms==0.1.0 && \
+    fix-permissions "${CONDA_DIR}"
diff --git a/docker/tinyms/0.1.0/Dockerfile b/docker/tinyms/0.1.0/Dockerfile
new file mode 100644
index 00000000..44cf7315
--- /dev/null
+++ b/docker/tinyms/0.1.0/Dockerfile
@@ -0,0 +1,10 @@
+ARG BASE_CONTAINER=mindspore/mindspore-cpu:1.1.1
+FROM $BASE_CONTAINER
+
+LABEL MAINTAINER="Leon Wang <wanghui71leon@gmail.com>"
+
+# Install base tools
+RUN apt-get update
+
+# Install TinyMS cpu whl package
+RUN pip install --no-cache-dir tinyms==0.1.0