diff --git a/docker/docker-npu/Dockerfile b/docker/docker-npu/Dockerfile
index dc35de47..3075a023 100644
--- a/docker/docker-npu/Dockerfile
+++ b/docker/docker-npu/Dockerfile
@@ -1,7 +1,7 @@
-# Use the Ubuntu 22.04 image with CANN 8.0.rc1
+# Use the Ubuntu 20.04 image with CANN 8.0.rc1
 # More versions can be found at https://hub.docker.com/r/ascendai/cann/tags
-# FROM ascendai/cann:8.0.rc1-910-ubuntu22.04-py3.8
-FROM ascendai/cann:8.0.rc1-910b-ubuntu22.04-py3.8
+# FROM ascendai/cann:8.0.rc1-910-ubuntu20.04-py3.8
+FROM ascendai/cann:8.0.rc1-910b-ubuntu20.04-py3.8
 # FROM ascendai/cann:8.0.rc1-910-openeuler22.03-py3.8
 # FROM ascendai/cann:8.0.rc1-910b-openeuler22.03-py3.8
 
@@ -10,7 +10,7 @@ ENV DEBIAN_FRONTEND=noninteractive
 
 # Define installation arguments
 ARG INSTALL_DEEPSPEED=false
-ARG PIP_INDEX=https://pypi.org/simple
+ARG PIP_INDEX=https://pypi.tuna.tsinghua.edu.cn/simple
 ARG TORCH_INDEX=https://download.pytorch.org/whl/cpu
 
 # Set the working directory