Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-07-09 03:04:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Dockerfile for rknn

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/docker.yml vendored
View File

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

6
machine-learning/Dockerfile
View File

@ -15,6 +15,8 @@ RUN mkdir /opt/armnn && \
cd /opt/ann && \ cd /opt/ann && \
sh build.sh sh build.sh
FROM builder-cpu AS builder-rknn
FROM builder-${DEVICE} AS builder FROM builder-${DEVICE} AS builder
ARG DEVICE ARG DEVICE
@ -80,6 +82,10 @@ COPY --from=builder-armnn \
/opt/ann/build.sh \ /opt/ann/build.sh \
/opt/armnn/ /opt/armnn/
FROM prod-cpu AS prod-rknn
ADD https://github.com/airockchip/rknn-toolkit2/raw/refs/tags/v2.3.0/rknpu2/runtime/Linux/librknn_api/aarch64/librknnrt.so /usr/lib/
FROM prod-${DEVICE} AS prod FROM prod-${DEVICE} AS prod
ARG DEVICE ARG DEVICE