[ci] update the docker image name (#1017)

docker/Dockerfile

@@ -1,9 +1,4 @@
-FROM nvcr.io/nvidia/pytorch:21.07-py3
-
-# install dependencies
-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
+FROM hpcaitech/pytorch-cuda:1.10.1-11.3.0
 
 # install colossalai
 RUN git clone https://github.com/hpcaitech/ColossalAI.git \