Merge pull request #5230 from MengqingCao/image

[NPU] Update npu base image
hiyouga · Aug 21, 2024 · 36039b0 · 36039b0
2 parents 8907150 + b3f4acd
commit 36039b0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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