diff --git a/cann/openeuler/Dockerfile b/cann/openeuler/Dockerfile
index d7285972..f2523763 100644
--- a/cann/openeuler/Dockerfile
+++ b/cann/openeuler/Dockerfile
@@ -1,12 +1,16 @@
 # Arguments
 ARG BASE_VERSION=latest
-ARG TARGETPLATFORM
+ARG PLATFORM=${TARGETPLATFORM}
 ARG CANN_CHIP
 ARG CANN_VERSION
 
 # Phase 1: download CANN installers
 FROM alpine as downloader
 
+ARG PLATFORM
+ARG CANN_CHIP
+ARG CANN_VERSION
+
 RUN apk add --no-cache \
         bash \
         curl \
@@ -18,9 +22,6 @@ COPY ../scripts/cann.sh /tmp/cann.sh
 
 # Download CANN
 RUN chmod +x /tmp/cann.sh && \
-    PLATFORM=${TARGETPLATFORM} \
-    CANN_CHIP=${CANN_CHIP} \
-    CANN_VERSION=${CANN_VERSION} \
     bash /tmp/cann.sh --download
 
 # Phase 2: install CANN toolkit and kernels
@@ -53,9 +54,6 @@ RUN yum install -y \
 RUN --mount=type=cache,from=downloader,target=/tmp,source=/tmp \
     umask 0022 && \
     chmod +x /tmp/cann.sh && \
-    PLATFORM=${TARGETPLATFORM} \
-    CANN_CHIP=${CANN_CHIP} \
-    CANN_VERSION=${CANN_VERSION} \
     bash /tmp/cann.sh --install && \
     rm -rf /tmp/*
 
diff --git a/cann/ubuntu/Dockerfile b/cann/ubuntu/Dockerfile
index 3fafbf20..bb111e9a 100644
--- a/cann/ubuntu/Dockerfile
+++ b/cann/ubuntu/Dockerfile
@@ -1,12 +1,16 @@
 # Arguments
 ARG BASE_VERSION=latest
-ARG TARGETPLATFORM
+ARG PLATFORM=${TARGETPLATFORM}
 ARG CANN_CHIP
 ARG CANN_VERSION
 
 # Phase 1: download CANN installers
 FROM alpine as downloader
 
+ARG PLATFORM
+ARG CANN_CHIP
+ARG CANN_VERSION
+
 RUN apk add --no-cache \
         bash \
         curl \
@@ -18,14 +22,15 @@ COPY ../scripts/cann.sh /tmp/cann.sh
 
 # Download CANN
 RUN chmod +x /tmp/cann.sh && \
-    PLATFORM=${TARGETPLATFORM} \
-    CANN_CHIP=${CANN_CHIP} \
-    CANN_VERSION=${CANN_VERSION} \
     bash /tmp/cann.sh --download
 
 # Phase 2: install CANN toolkit and kernels
 FROM ubuntu:${BASE_VERSION} as builder
 
+ARG PLATFORM
+ARG CANN_CHIP
+ARG CANN_VERSION
+
 # Change the default shell
 SHELL [ "/bin/bash", "-c" ]
 
@@ -61,18 +66,15 @@ RUN apt-get update \
         python3-pip \
         python3-dev \
         vim \
-     && apt-get clean \
-     && rm -rf /var/lib/apt/lists/* \
-     && rm -rf /var/tmp/* \
-     && rm -rf /tmp/*
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/* \
+    && rm -rf /var/tmp/* \
+    && rm -rf /tmp/*
 
 # Install CANN
 RUN --mount=type=cache,from=downloader,target=/tmp,source=/tmp \
     umask 0022 && \
     chmod +x /tmp/cann.sh && \
-    PLATFORM=${TARGETPLATFORM} \
-    CANN_CHIP=${CANN_CHIP} \
-    CANN_VERSION=${CANN_VERSION} \
     bash /tmp/cann.sh --install && \
     rm -rf /tmp/*