From e4a79b12ebbe47fba61990cc5bc7853361de2e73 Mon Sep 17 00:00:00 2001 From: Blake Blackshear Date: Sat, 31 Dec 2022 09:13:08 -0600 Subject: [PATCH] set env vars needed for nvidia support (#4837) --- Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile b/Dockerfile index 4ea2997fa..7b95d5828 100644 --- a/Dockerfile +++ b/Dockerfile @@ -208,6 +208,7 @@ ARG DEBIAN_FRONTEND ARG APT_KEY_DONT_WARN_ON_DANGEROUS_USAGE=DontWarn # https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(Native-GPU-Support) +ENV NVIDIA_VISIBLE_DEVICES=all ENV NVIDIA_DRIVER_CAPABILITIES="compute,video,utility" ENV PATH="/usr/lib/btbn-ffmpeg/bin:/usr/local/go2rtc/bin:/usr/local/nginx/sbin:${PATH}"