Paddle ocr gpu support. #4
@@ -25,6 +25,7 @@ jobs:
|
|||||||
echo "Version: 1.0.${{ gitea.run_number }}" >> $GITHUB_STEP_SUMMARY
|
echo "Version: 1.0.${{ gitea.run_number }}" >> $GITHUB_STEP_SUMMARY
|
||||||
echo "Event: ${{ gitea.event_name }}" >> $GITHUB_STEP_SUMMARY
|
echo "Event: ${{ gitea.event_name }}" >> $GITHUB_STEP_SUMMARY
|
||||||
|
|
||||||
|
# CPU image: Multi-arch (amd64 + arm64)
|
||||||
build_cpu:
|
build_cpu:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
needs: essential
|
needs: essential
|
||||||
@@ -32,25 +33,31 @@ jobs:
|
|||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
|
- name: Set up QEMU
|
||||||
|
uses: docker/setup-qemu-action@v3
|
||||||
|
|
||||||
|
- name: Set up Docker Buildx
|
||||||
|
uses: docker/setup-buildx-action@v3
|
||||||
|
|
||||||
- name: Login to Gitea Registry
|
- name: Login to Gitea Registry
|
||||||
run: |
|
uses: docker/login-action@v3
|
||||||
echo ${{ secrets.UNIR_REGISTRY_TOKEN }} | docker login \
|
with:
|
||||||
-u username \
|
registry: ${{ needs.essential.outputs.repo }}
|
||||||
--password-stdin ${{ needs.essential.outputs.repo }}
|
username: username
|
||||||
|
password: ${{ secrets.UNIR_REGISTRY_TOKEN }}
|
||||||
|
|
||||||
- name: Build CPU image
|
- name: Build and push CPU image (multi-arch)
|
||||||
run: |
|
uses: docker/build-push-action@v5
|
||||||
docker build \
|
with:
|
||||||
-f src/paddle_ocr/Dockerfile.cpu \
|
context: src/paddle_ocr
|
||||||
-t ${{ needs.essential.outputs.image_cpu }}:${{ needs.essential.outputs.Version }} \
|
file: src/paddle_ocr/Dockerfile.cpu
|
||||||
-t ${{ needs.essential.outputs.image_cpu }}:latest \
|
platforms: linux/amd64,linux/arm64
|
||||||
src/paddle_ocr/
|
push: true
|
||||||
|
tags: |
|
||||||
- name: Push CPU image
|
${{ needs.essential.outputs.image_cpu }}:${{ needs.essential.outputs.Version }}
|
||||||
run: |
|
${{ needs.essential.outputs.image_cpu }}:latest
|
||||||
docker push ${{ needs.essential.outputs.image_cpu }}:${{ needs.essential.outputs.Version }}
|
|
||||||
docker push ${{ needs.essential.outputs.image_cpu }}:latest
|
|
||||||
|
|
||||||
|
# GPU image: x86_64 only (PaddlePaddle GPU doesn't support ARM64)
|
||||||
build_gpu:
|
build_gpu:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
needs: essential
|
needs: essential
|
||||||
@@ -64,7 +71,7 @@ jobs:
|
|||||||
-u username \
|
-u username \
|
||||||
--password-stdin ${{ needs.essential.outputs.repo }}
|
--password-stdin ${{ needs.essential.outputs.repo }}
|
||||||
|
|
||||||
- name: Build GPU image
|
- name: Build GPU image (x86_64)
|
||||||
run: |
|
run: |
|
||||||
docker build \
|
docker build \
|
||||||
-f src/paddle_ocr/Dockerfile.gpu \
|
-f src/paddle_ocr/Dockerfile.gpu \
|
||||||
|
|||||||
Reference in New Issue
Block a user