diff --git a/docker/Dockerfile b/docker/Dockerfile
new file mode 100644
index 000000000..a58e8dc61
--- /dev/null
+++ b/docker/Dockerfile
@@ -0,0 +1,10 @@
+FROM nvcr.io/nvidia/pytorch:21.07-py3
+
+# install dependencies
+RUN pip install -U pip setuptools wheel \
+ && pip install pytest tensorboard deepspeed apex
+
+# install colossalai
+RUN git clone https://github.com/hpcaitech/ColossalAI.git \
+ && cd ./ColossalAI \
+ && pip install -v --no-cache-dir --global-option="--cuda_ext" .