Fix errors when using nvidia images

blakeblackshear · Nov 17, 2020 · 777aff4 · 777aff4
1 parent 4b3b702
commit 777aff4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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