Paddle ocr gpu support. #4

Merged
Seryusjj merged 40 commits from gpu_support into main 2026-01-19 17:35:25 +00:00
Showing only changes of commit e8e7a61a60 - Show all commits

View File

@@ -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 \