Add frigate-jetson image with hwaccel ffmpeg codecs

This commit is contained in:
Andrew Reiter 2023-05-02 14:05:16 -04:00
parent b38c9e82e2
commit e4306a3c05
3 changed files with 28 additions and 0 deletions

View File

@ -65,3 +65,13 @@ jobs:
tags: | tags: |
ghcr.io/${{ steps.lowercaseRepo.outputs.lowercase }}:${{ github.ref_name }}-${{ env.SHORT_SHA }}-tensorrt ghcr.io/${{ steps.lowercaseRepo.outputs.lowercase }}:${{ github.ref_name }}-${{ env.SHORT_SHA }}-tensorrt
cache-from: type=gha cache-from: type=gha
- name: Build and push Jetson build
uses: docker/build-push-action@v4
with:
context: .
push: true
platforms: linux/arm64
target: frigate-jetson
tags: |
ghcr.io/${{ steps.lowercaseRepo.outputs.lowercase }}:${{ github.ref_name }}-${{ env.SHORT_SHA }}-tensorrt
cache-from: type=gha

View File

@ -260,6 +260,11 @@ RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels
ln -s libnvrtc.so.11.2 /usr/local/lib/python3.9/dist-packages/nvidia/cuda_nvrtc/lib/libnvrtc.so && \ ln -s libnvrtc.so.11.2 /usr/local/lib/python3.9/dist-packages/nvidia/cuda_nvrtc/lib/libnvrtc.so && \
ldconfig ldconfig
# NVIDIA Jetson platforms have their own codecs too
FROM frigate-tensorrt AS frigate-jetson
RUN --mount=type=bind,source=docker/install_jetson_ffmpeg.sh,target=/deps/install_jetson_ffmpeg.sh \
/deps/install_jetson_ffmpeg.sh
# Dev Container w/ TRT # Dev Container w/ TRT
FROM devcontainer AS devcontainer-trt FROM devcontainer AS devcontainer-trt

13
docker/install_jetson_ffmpeg.sh Executable file
View File

@ -0,0 +1,13 @@
#!/bin/bash
set -euxo pipefail
if [[ "${TARGETARCH}" == "arm64" ]]; then
echo "deb https://repo.download.nvidia.com/jetson/ffmpeg main main" | tee /etc/apt/sources.list.d/jetson_ffmpeg.list
echo "deb-src https://repo.download.nvidia.com/jetson/ffmpeg main main" | tee /etc/apt/sources.list/jetson_ffmpeg.list
apt-get -qq update
apt-get -qq install --no-install-recommends --no-install-suggests -y ffmpeg
fi
rm -rf /var/lib/apt/lists/*