From 8d06186ff93552bacbb875517daaea002cec922b Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Tue, 24 May 2022 17:51:50 +0800
Subject: [PATCH] [doc] update docker instruction (#1020)

---
 README-zh-Hans.md | 3 +++
 README.md         | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/README-zh-Hans.md b/README-zh-Hans.md
index 569a5208f..6d69dda14 100644
--- a/README-zh-Hans.md
+++ b/README-zh-Hans.md
@@ -178,6 +178,9 @@ NO_CUDA_EXT=1 pip install .
 
 运行以下命令从我们提供的 docker 文件中建立 docker 镜像。
 
+> 在Dockerfile里编译Colossal-AI需要有GPU支持，您需要将Nvidia Docker Runtime设置为默认的Runtime。更多信息可以点击[这里](https://stackoverflow.com/questions/59691207/docker-build-with-nvidia-runtime)。
+> 我们推荐从[项目主页](https://www.colossalai.org)直接下载Colossal-AI.
+
 ```bash
 cd ColossalAI
 docker build -t colossalai ./docker
diff --git a/README.md b/README.md
index 85faf5095..eb069a382 100644
--- a/README.md
+++ b/README.md
@@ -183,6 +183,9 @@ NO_CUDA_EXT=1 pip install .
 
 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