From 51f1ec96b0140f617749a48226d092018ca22434 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Thu, 23 Jun 2022 15:12:15 +0800
Subject: [PATCH] [workflow] polish readme and dockerfile (#1165)

* [workflow] polish readme and dockerfile

* polish
---
 README-zh-Hans.md | 6 ++++++
 README.md         | 8 ++++++++
 docker/Dockerfile | 3 +++
 3 files changed, 17 insertions(+)

diff --git a/README-zh-Hans.md b/README-zh-Hans.md
index 16cdd0ca6..27b966096 100644
--- a/README-zh-Hans.md
+++ b/README-zh-Hans.md
@@ -196,6 +196,12 @@ NO_CUDA_EXT=1 pip install .
 
 ## 使用 Docker
 
+### 从DockerHub获取镜像
+
+您可以直接从我们的[DockerHub主页](https://hub.docker.com/r/hpcaitech/colossalai)获取最新的镜像，每一次发布我们都会自动上传最新的镜像。
+
+### 本地构建镜像
+
 运行以下命令从我们提供的 docker 文件中建立 docker 镜像。
 
 > 在Dockerfile里编译Colossal-AI需要有GPU支持，您需要将Nvidia Docker Runtime设置为默认的Runtime。更多信息可以点击[这里](https://stackoverflow.com/questions/59691207/docker-build-with-nvidia-runtime)。
diff --git a/README.md b/README.md
index 05dba1ef3..8fe1e7097 100644
--- a/README.md
+++ b/README.md
@@ -201,11 +201,19 @@ NO_CUDA_EXT=1 pip install .
 
 ## Use Docker
 
+### Pull from DockerHub
+
+You can directly pull the docker image from our [DockerHub page](https://hub.docker.com/r/hpcaitech/colossalai). The image is automatically uploaded upon release.
+
+
+### Build On Your Own
+
 Run the following command to build a docker image from Dockerfile provided.
 
 > Building Colossal-AI from scratch requires GPU support, you need to use Nvidia Docker Runtime as the default when doing `docker build`. More details can be found [here](https://stackoverflow.com/questions/59691207/docker-build-with-nvidia-runtime).
 > We recommend you install Colossal-AI from our [project page](https://www.colossalai.org) directly.
 
+
 ```bash
 cd ColossalAI
 docker build -t colossalai ./docker
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 7dd82fc93..2ccd3ad66 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -12,3 +12,6 @@ RUN git clone https://github.com/NVIDIA/apex && \
 RUN git clone https://github.com/hpcaitech/ColossalAI.git \
  && cd ./ColossalAI \
  && pip install -v --no-cache-dir .
+
+# install titans
+RUN pip install -no-cache-dir titans