From 10bdef414eb82c1900408dac97b9709efc72c415 Mon Sep 17 00:00:00 2001
From: Koushik Dutta <koushd@gmail.com>
Date: Wed, 9 Oct 2024 13:06:54 -0700
Subject: [PATCH] install: auto enable nvidia gpus and capabilities

---
 install/docker/Dockerfile.nvidia | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/install/docker/Dockerfile.nvidia b/install/docker/Dockerfile.nvidia
index 67c81493f..6cb9265e7 100644
--- 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