From 75312602aa2bcaa2df4ce58cbfd958b4fa28357c Mon Sep 17 00:00:00 2001
From: Blake Blackshear <blakeb@blakeshome.com>
Date: Sun, 25 Oct 2020 12:15:50 -0500
Subject: [PATCH] add support for iHD driver

---
 docker/Dockerfile.amd64 | 4 +++-
 docker/Dockerfile.base  | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/docker/Dockerfile.amd64 b/docker/Dockerfile.amd64
index ac3100869..cd1e262ed 100644
--- a/docker/Dockerfile.amd64
+++ b/docker/Dockerfile.amd64
@@ -1,13 +1,15 @@
 FROM frigate-base
 LABEL maintainer "blakeb@blakeshome.com"
 
+# By default, use the i965 driver
+ENV LIBVA_DRIVER_NAME=i965
 # Install packages for apt repo
 RUN apt-get -qq update \
     && apt-get -qq install --no-install-recommends -y \
         # ffmpeg dependencies
         libgomp1 \
         # VAAPI drivers for Intel hardware accel
-        libva-drm2 libva2 i965-va-driver vainfo \
+        libva-drm2 libva2 i965-va-driver vainfo intel-media-va-driver \
         ## Tensorflow lite
     && wget -q https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
     && python3.8 -m pip install tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
diff --git a/docker/Dockerfile.base b/docker/Dockerfile.base
index d19644cfe..a9b26bfa5 100644
--- a/docker/Dockerfile.base
+++ b/docker/Dockerfile.base
@@ -12,7 +12,9 @@ COPY --from=wheels /wheels/. /wheels/
 ENV FLASK_ENV=development
 ENV DEBIAN_FRONTEND=noninteractive
 # Install packages for apt repo
-RUN apt-get -qq update && apt-get -qq install --no-install-recommends -y \
+RUN apt-get -qq update \
+    && apt-get upgrade -y \
+    && apt-get -qq install --no-install-recommends -y \
     gnupg wget unzip tzdata \
     && apt-get -qq install --no-install-recommends -y \
         python3-pip \