diff --git a/docker/docker-npu/Dockerfile b/docker/docker-npu/Dockerfile
index 34cf9616..dc35de47 100644
--- a/docker/docker-npu/Dockerfile
+++ b/docker/docker-npu/Dockerfile
@@ -1,9 +1,9 @@
 # Use the Ubuntu 22.04 image with CANN 8.0.rc1
-# More versions can be found at https://hub.docker.com/r/cosdt/cann/tags
-# FROM cosdt/cann:8.0.rc1-910-ubuntu22.04
-FROM cosdt/cann:8.0.rc1-910b-ubuntu22.04
-# FROM cosdt/cann:8.0.rc1-910-openeuler22.03
-# FROM cosdt/cann:8.0.rc1-910b-openeuler22.03
+# 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-openeuler22.03-py3.8
+# FROM ascendai/cann:8.0.rc1-910b-openeuler22.03-py3.8
 
 # Define environments
 ENV DEBIAN_FRONTEND=noninteractive