Dockerfile for rknn

This commit is contained in:
yoni13
2025-01-11 15:47:24 +08:00
parent d5ef821b24
commit 506ca0d3a4
2 changed files with 10 additions and 1 deletions

View File

@@ -48,7 +48,7 @@ jobs:
runs-on: ubuntu-latest
strategy:
matrix:
suffix: ["", "-cuda", "-openvino", "-armnn"]
suffix: ["", "-cuda", "-openvino", "-armnn","-rknn"]
steps:
- name: Login to GitHub Container Registry
uses: docker/login-action@v3
@@ -116,6 +116,9 @@ jobs:
- platforms: linux/arm64
device: armnn
suffix: -armnn
- platforms: linux/arm64
device: rknn
suffix: -rknn
steps:
- name: Checkout