mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-29 20:23:51 -06:00
48 lines
2.2 KiB
Docker
48 lines
2.2 KiB
Docker
FROM frigate-base
|
|
LABEL maintainer "blakeb@blakeshome.com"
|
|
|
|
# Install packages for apt repo
|
|
RUN apt-get -qq update \
|
|
&& apt-get -qq install --no-install-recommends -y \
|
|
# ffmpeg dependencies
|
|
libgomp1 \
|
|
## Tensorflow lite
|
|
&& wget -q https://github.com/google-coral/pycoral/releases/download/release-frogfish/tflite_runtime-2.5.0-cp38-cp38-linux_x86_64.whl \
|
|
&& python3.8 -m pip install tflite_runtime-2.5.0-cp38-cp38-linux_x86_64.whl \
|
|
&& rm tflite_runtime-2.5.0-cp38-cp38-linux_x86_64.whl \
|
|
&& rm -rf /var/lib/apt/lists/* \
|
|
&& (apt-get autoremove -y; apt-get autoclean -y)
|
|
|
|
|
|
# nvidia layer (see https://gitlab.com/nvidia/container-images/cuda/blob/master/dist/11.1/ubuntu20.04-x86_64/base/Dockerfile)
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,video
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
gnupg2 curl ca-certificates && \
|
|
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub | apt-key add - && \
|
|
echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/cuda.list && \
|
|
echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list && \
|
|
apt-get purge --autoremove -y curl \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
ENV CUDA_VERSION 11.1.1
|
|
|
|
# For libraries in the cuda-compat-* package: https://docs.nvidia.com/cuda/eula/index.html#attachment-a
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
cuda-cudart-11-1=11.1.74-1 \
|
|
cuda-compat-11-1 \
|
|
&& ln -s cuda-11.1 /usr/local/cuda && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Required for nvidia-docker v1
|
|
RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
|
|
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
|
|
|
|
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
|
|
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
|
|
|
|
# nvidia-container-runtime
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,video
|
|
ENV NVIDIA_REQUIRE_CUDA "cuda>=11.1 brand=tesla,driver>=418,driver<419 brand=tesla,driver>=440,driver<441 brand=tesla,driver>=450,driver<451"
|