From 777aff403f5d0db548d2ceb83a0667030477a1f9 Mon Sep 17 00:00:00 2001 From: Gerard Escalante Date: Sun, 15 Nov 2020 06:10:50 +0000 Subject: [PATCH] Fix errors when using nvidia images --- Makefile | 2 +- docker/Dockerfile.amd64nvidia | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index d1f1838e7..7b6789ff8 100644 --- a/Makefile +++ b/Makefile @@ -16,7 +16,7 @@ amd64nvidia_wheels: docker build --tag blakeblackshear/frigate-wheels:amd64nvidia --file docker/Dockerfile.wheels . amd64nvidia_ffmpeg: - docker build --tag blakeblackshear/frigate-ffmpeg:amd64nvidia --file docker/Dockerfile.ffmpeg.amd64 . + docker build --tag blakeblackshear/frigate-ffmpeg:amd64nvidia --file docker/Dockerfile.ffmpeg.amd64nvidia . amd64nvidia_frigate: docker build --tag frigate-base --build-arg ARCH=amd64nvidia --file docker/Dockerfile.base . diff --git a/docker/Dockerfile.amd64nvidia b/docker/Dockerfile.amd64nvidia index 7d63a04dc..a3c7e612d 100644 --- a/docker/Dockerfile.amd64nvidia +++ b/docker/Dockerfile.amd64nvidia @@ -31,6 +31,7 @@ ENV CUDA_VERSION 11.1.1 RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-cudart-11-1=11.1.74-1 \ cuda-compat-11-1 \ + libnvidia-decode-455 \ && ln -s cuda-11.1 /usr/local/cuda && \ rm -rf /var/lib/apt/lists/* @@ -39,7 +40,7 @@ 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 +ENV LD_LIBRARY_PATH /usr/local/cuda/lib64 # nvidia-container-runtime ENV NVIDIA_VISIBLE_DEVICES all