install: auto enable nvidia gpus and capabilities

koush · Oct 9, 2024 · 10bdef4 · 10bdef4
1 parent 317ee80
commit 10bdef4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/install/docker/Dockerfile.nvidia b/install/docker/Dockerfile.nvidia
@@ -1,6 +1,9 @@
 ARG BASE="ghcr.io/koush/scrypted-common:20-jammy-full"
 FROM $BASE
 
+ENV NVIDIA_DRIVER_CAPABILITIES=all
+ENV NVIDIA_VISIBLE_DEVICES=all
+
 # nvidia cudnn/libcublas etc.
 # for some reason this is not provided by the nvidia container toolkit
 RUN curl https://raw.githubusercontent.com/koush/scrypted/main/install/docker/install-nvidia-graphics.sh | bash