From 083eba1dfe8c3b1d540db23ea4fa3c23d7be5081 Mon Sep 17 00:00:00 2001 From: Nate Meyer Date: Fri, 30 Dec 2022 11:00:34 -0500 Subject: [PATCH] Add CI stage to build TensorRT tag --- .github/workflows/ci.yml | 12 ++++++++++++ requirements-wheels.txt | 2 +- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index c89b4e9ba..1c166eb10 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -36,7 +36,19 @@ jobs: context: . push: true platforms: linux/amd64,linux/arm64,linux/arm/v7 + target: frigate tags: | ghcr.io/blakeblackshear/frigate:${{ github.ref_name }}-${{ env.SHORT_SHA }} cache-from: type=gha cache-to: type=gha,mode=max + - name: Build and push TensorRT + uses: docker/build-push-action@v3 + with: + context: . + push: true + platforms: linux/amd64 + target: frigate-tensorrt + tags: | + ghcr.io/blakeblackshear/frigate:${{ github.ref_name }}-${{ env.SHORT_SHA }}-tensorrt + cache-from: type=gha + cache-to: type=gha,mode=max \ No newline at end of file diff --git a/requirements-wheels.txt b/requirements-wheels.txt index 2c82a50fc..930ab9e47 100644 --- a/requirements-wheels.txt +++ b/requirements-wheels.txt @@ -23,4 +23,4 @@ zeroconf == 0.39.4 # Openvino Library - Custom built with MYRIAD support openvino @ https://github.com/NateMeyer/openvino-wheels/releases/download/multi-arch_2022.2.0/openvino-2022.2.0-000-cp39-cp39-manylinux_2_31_x86_64.whl; platform_machine == 'x86_64' openvino @ https://github.com/NateMeyer/openvino-wheels/releases/download/multi-arch_2022.2.0/openvino-2022.2.0-000-cp39-cp39-linux_aarch64.whl; platform_machine == 'aarch64' -openvino @ https://github.com/NateMeyer/openvino-wheels/releases/download/multi-arch_2022.2.0/openvino-2022.2.0-000-cp39-cp39-linux_armv7l.whl; platform_machine == 'armv7l' \ No newline at end of file +openvino @ https://github.com/NateMeyer/openvino-wheels/releases/download/multi-arch_2022.2.0/openvino-2022.2.0-000-cp39-cp39-linux_armv7l.whl; platform_machine == 'armv7l'