mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-26 19:06:11 +01:00
dd02958f7c
* Update to latest tensorrt (8.6.1) release * Build trt libyolo_layer.so in container * Update tensorrt_models script to convert models from the frigate container * Fix typo in model script * Fix paths to yolo lib and models folder * Add S6 scripts to test and convert specified TensortRT models at startup. Rearrange tensorrt files into a docker support folder. * Update TensorRT documentation to reflect the new model conversion process and minimum HW support. * Fix model_cache path to live in config directory * Move tensorrt s6 files to the correct directory * Fix issues in model generation script * Disable global timeout for s6 services * Add version folder to tensorrt model_cache path * Include TensorRT version 8.5.3 * Add numpy requirement prior to removal of np.bool * This TRT version uses a mixture of cuda dependencies * Redirect stdout from noisy model conversion
81 lines
3.1 KiB
Bash
Executable File
81 lines
3.1 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euxo pipefail
|
|
|
|
apt-get -qq update
|
|
|
|
apt-get -qq install --no-install-recommends -y \
|
|
apt-transport-https \
|
|
gnupg \
|
|
wget \
|
|
procps vainfo \
|
|
unzip locales tzdata libxml2 xz-utils \
|
|
python3-pip \
|
|
curl \
|
|
jq \
|
|
nethogs
|
|
|
|
mkdir -p -m 600 /root/.gnupg
|
|
|
|
# add coral repo
|
|
curl -fsSLo - https://packages.cloud.google.com/apt/doc/apt-key.gpg | \
|
|
gpg --dearmor -o /etc/apt/trusted.gpg.d/google-cloud-packages-archive-keyring.gpg
|
|
echo "deb https://packages.cloud.google.com/apt coral-edgetpu-stable main" | tee /etc/apt/sources.list.d/coral-edgetpu.list
|
|
echo "libedgetpu1-max libedgetpu/accepted-eula select true" | debconf-set-selections
|
|
|
|
# enable non-free repo
|
|
sed -i -e's/ main/ main contrib non-free/g' /etc/apt/sources.list
|
|
|
|
# coral drivers
|
|
apt-get -qq update
|
|
apt-get -qq install --no-install-recommends --no-install-suggests -y \
|
|
libedgetpu1-max python3-tflite-runtime python3-pycoral
|
|
|
|
# btbn-ffmpeg -> amd64
|
|
if [[ "${TARGETARCH}" == "amd64" ]]; then
|
|
mkdir -p /usr/lib/btbn-ffmpeg
|
|
wget -qO btbn-ffmpeg.tar.xz "https://github.com/BtbN/FFmpeg-Builds/releases/download/autobuild-2022-07-31-12-37/ffmpeg-n5.1-2-g915ef932a3-linux64-gpl-5.1.tar.xz"
|
|
tar -xf btbn-ffmpeg.tar.xz -C /usr/lib/btbn-ffmpeg --strip-components 1
|
|
rm -rf btbn-ffmpeg.tar.xz /usr/lib/btbn-ffmpeg/doc /usr/lib/btbn-ffmpeg/bin/ffplay
|
|
fi
|
|
|
|
# ffmpeg -> arm64
|
|
if [[ "${TARGETARCH}" == "arm64" ]]; then
|
|
# add raspberry pi repo
|
|
gpg --no-default-keyring --keyring /usr/share/keyrings/raspbian.gpg --keyserver keyserver.ubuntu.com --recv-keys 82B129927FA3303E
|
|
echo "deb [signed-by=/usr/share/keyrings/raspbian.gpg] https://archive.raspberrypi.org/debian/ bullseye main" | tee /etc/apt/sources.list.d/raspi.list
|
|
apt-get -qq update
|
|
apt-get -qq install --no-install-recommends --no-install-suggests -y ffmpeg
|
|
fi
|
|
|
|
# arch specific packages
|
|
if [[ "${TARGETARCH}" == "amd64" ]]; then
|
|
# Use debian testing repo only for hwaccel packages
|
|
echo 'deb http://deb.debian.org/debian testing main non-free' >/etc/apt/sources.list.d/debian-testing.list
|
|
apt-get -qq update
|
|
# intel-opencl-icd specifically for GPU support in OpenVino
|
|
apt-get -qq install --no-install-recommends --no-install-suggests -y \
|
|
intel-opencl-icd \
|
|
mesa-va-drivers libva-drm2 intel-media-va-driver-non-free i965-va-driver libmfx1 radeontop intel-gpu-tools
|
|
# something about this dependency requires it to be installed in a separate call rather than in the line above
|
|
apt-get -qq install --no-install-recommends --no-install-suggests -y \
|
|
i965-va-driver-shaders
|
|
rm -f /etc/apt/sources.list.d/debian-testing.list
|
|
fi
|
|
|
|
if [[ "${TARGETARCH}" == "arm64" ]]; then
|
|
apt-get -qq install --no-install-recommends --no-install-suggests -y \
|
|
libva-drm2 mesa-va-drivers
|
|
fi
|
|
|
|
apt-get purge gnupg apt-transport-https xz-utils -y
|
|
apt-get clean autoclean -y
|
|
apt-get autoremove --purge -y
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install yq, for frigate-prepare and go2rtc echo source
|
|
curl -fsSL \
|
|
"https://github.com/mikefarah/yq/releases/download/v4.33.3/yq_linux_$(dpkg --print-architecture)" \
|
|
--output /usr/local/bin/yq
|
|
chmod +x /usr/local/bin/yq
|