From a1da3900c8b7bf85844e134e65ef2d07ed9c91a7 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Tue, 18 Jan 2022 13:35:18 +0800
Subject: [PATCH] added docker documentation (#152)

---
 README.md         | 16 ++++++++++++++++
 docker/Dockerfile |  3 ++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 1d8b81e98..3f8dfc556 100644
--- a/README.md
+++ b/README.md
@@ -38,6 +38,22 @@ pip install -v --no-cache-dir --global-option="--cuda_ext" .
 
 - [Documentation](https://www.colossalai.org/)
 
+
+## Use Docker
+
+Run the following command to build a docker image from Dockerfile provided.
+
+```bash
+cd ColossalAI
+docker build -t colossalai ./docker
+```
+
+Run the following command to start the docker container in interactive mode.
+
+```bash
+docker run -ti --gpus all --rm --ipc=host colossalai bash
+```
+
 ## Quick View
 
 ### Start Distributed Training in Lines
diff --git a/docker/Dockerfile b/docker/Dockerfile
index a58e8dc61..f758e984b 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,7 +1,8 @@
 FROM nvcr.io/nvidia/pytorch:21.07-py3
 
 # install dependencies
-RUN pip install -U pip setuptools wheel \
+RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple \
+ && pip install -U pip setuptools wheel \
  && pip install pytest tensorboard deepspeed apex
 
 # install colossalai