add volces cr

dptech-corp · Jun 24, 2024 · aa98b8f · aa98b8f
1 parent 3ee5bba
commit aa98b8f
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 11 deletions.
diff --git a/.github/workflows/docker_rdma.yml b/.github/workflows/docker_rdma.yml
@@ -14,16 +14,23 @@ jobs:
         uses: actions/checkout@v3
       -
         name: Set up QEMU
-        uses: docker/setup-qemu-action@v2
+        uses: docker/setup-qemu-action@v3
       -
         name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
+        uses: docker/setup-buildx-action@v3
       -
         name: Login to DockerHub
-        uses: docker/login-action@v2
+        uses: docker/login-action@v3
         with:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
+      -
+        name: Login to Volces CR
+        uses: docker/login-action@v3
+        with:
+          registry: dp-ve-registry-cn-beijing.cr.volces.com
+          username: ${{ secrets.VOLCES_CR_USERNAME }}
+          password: ${{ secrets.VOLCES_CR_TOKEN }}
       -
         name: Free Disk Space (Ubuntu)
         uses: jlumbroso/free-disk-space@main
@@ -47,8 +54,10 @@ jobs:
           swap-size-gb: 10
       -
         name: Build and push with rdma
-        uses: docker/build-push-action@v3
+        uses: docker/build-push-action@v6
         with:
           context: ./docker/rdma/
           push: true
-          tags: dptechnology/unicore:${{ github.ref_name }}-pytorch2.0.1-cuda11.7-rdma
+          tags: |
+            dptechnology/unicore:${{ github.ref_name }}-pytorch2.0.1-cuda11.7-rdma
+            dp-ve-registry-cn-beijing.cr.volces.com/dplc/unicore:${{ github.ref_name }}-pytorch2.0.1-cuda11.7-rdma
diff --git a/.github/workflows/docker_rdma_latest.yml b/.github/workflows/docker_rdma_latest.yml
@@ -1,10 +1,11 @@
-name: Build and Publish Docker
+name: Build and Publish Docker (latest)
 
 on:
   push:
     branches:
     - main
     - cuda12
+    - fix-docker-build
 
 jobs:
   docker:
@@ -15,16 +16,23 @@ jobs:
         uses: actions/checkout@v3
       -
         name: Set up QEMU
-        uses: docker/setup-qemu-action@v2
+        uses: docker/setup-qemu-action@v3
       -
         name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
+        uses: docker/setup-buildx-action@v3
       -
         name: Login to DockerHub
-        uses: docker/login-action@v2
+        uses: docker/login-action@v3
         with:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
+      -
+        name: Login to Volces CR
+        uses: docker/login-action@v3
+        with:
+          registry: dp-ve-registry-cn-beijing.cr.volces.com
+          username: ${{ secrets.VOLCES_CR_USERNAME }}
+          password: ${{ secrets.VOLCES_CR_TOKEN }}
       -
         name: Free Disk Space (Ubuntu)
         uses: jlumbroso/free-disk-space@main
@@ -48,8 +56,10 @@ jobs:
           swap-size-gb: 10
       -
         name: Build and push with rdma
-        uses: docker/build-push-action@v3
+        uses: docker/build-push-action@v6
         with:
           context: ./docker/rdma/
           push: true
-          tags: dptechnology/unicore:latest-pytorch2.3.0-cuda12.4-rdma
+          tags: |
+            dptechnology/unicore:latest-pytorch2.3.0-cuda12.4-rdma
+            dp-ve-registry-cn-beijing.cr.volces.com/dplc/unicore:latest-pytorch2.3.0-cuda12.4-rdma