diff --git a/.github/workflows/publish-arm-gpu-image.yaml b/.github/workflows/publish-arm-gpu-image.yaml
index c561cef5db77d..3a6d36b051c19 100644
--- a/.github/workflows/publish-arm-gpu-image.yaml
+++ b/.github/workflows/publish-arm-gpu-image.yaml
@@ -35,8 +35,7 @@ jobs:
         password: ${{ secrets.DOCKERHUB_TOKEN }}
     - name: docker build
       run: |
-        docker stop milvus && docker rm milvus  
-        docker run  --name milvus -v $(pwd):/root/milvus  -w /root/milvus  milvusdb/milvus-env:0.0.3-master-gpu -- make milvus-gpu
+        docker run -v $(pwd):/root/milvus  -w /root/milvus  milvusdb/milvus-env:0.0.4-master-gpu -- make milvus-gpu
     - name: docker package
       run: |
         docker build -t huangliang1203/milvus:${{ github.event.inputs.tagName }}-gpu-arm64 -f build/docker/milvus/gpu/ubuntu22.04/Dockerfile . --push