added docker documentation (#152)

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

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