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