From 22d0ce4f82f3e1aaefde275228ff24582cd38ebb Mon Sep 17 00:00:00 2001
From: Koushik Dutta <koushd@gmail.com>
Date: Thu, 10 Jul 2025 19:45:27 -0700
Subject: [PATCH] install: add nvidia to install script

---
 install/docker/docker-compose.yml             |  4 ++--
 .../docker/install-scrypted-docker-compose.sh | 21 +++++++++++++++++++
 server/package-lock.json                      |  4 ++--
 3 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/install/docker/docker-compose.yml b/install/docker/docker-compose.yml
index 5ef48e4cb..b91d4e0d4 100644
--- a/install/docker/docker-compose.yml
+++ b/install/docker/docker-compose.yml
@@ -45,10 +45,10 @@ services:
             # - SCRYPTED_DOCKER_AVAHI=true
 
         # NVIDIA (Part 1 of 2)
-        # runtime: nvidia
+        # nvidia runtime: nvidia
 
         # NVIDIA (Part 2 of 2) - Use NVIDIA image, and remove subsequent default image.
-        # image: ghcr.io/koush/scrypted:nvidia
+        # nvidia image: ghcr.io/koush/scrypted:nvidia
         image: ghcr.io/koush/scrypted
 
         volumes:
diff --git a/install/docker/install-scrypted-docker-compose.sh b/install/docker/install-scrypted-docker-compose.sh
index d982a9413..299cdf7c9 100755
--- a/install/docker/install-scrypted-docker-compose.sh
+++ b/install/docker/install-scrypted-docker-compose.sh
@@ -13,6 +13,8 @@ then
 fi
 
 function readyn() {
+    echo
+    echo
     if [ ! -z "$SCRYPTED_NONINTERACTIVE" ]
     then
         yn="y"
@@ -93,6 +95,25 @@ else
     sudo apt -y purge apparmor || true
 fi
 
+HAS_NVIDIA=$(lspci | grep -i nvidia)
+if [ ! -z "$HAS_NVIDIA" ]
+then
+    readyn "NVIDIA GPU detected. Use NVIDIA image for GPU acceleration?"
+    if [ "$yn" == "y" ]
+    then
+        readyn "NVIDIA image requires the NVIDIA Drivers and Container Toolkit to be installed. This script can install them for you. Install NVIDIA Drivers and Container Toolkit for GPU acceleration?"
+        if [ "$yn" == "y" ]
+        then
+            curl -fsSL https://raw.githubusercontent.com/koush/scrypted/main/install/docker/install-nvidia-container-toolkit.sh -o install-nvidia-container-toolkit.sh
+            chmod +x install-nvidia-container-toolkit.sh
+            ./install-nvidia-container-toolkit.sh
+            rm install-nvidia-container-toolkit.sh
+        fi
+        sed -i 's/'#' nvidia //g' $DOCKER_COMPOSE_YML
+        sed -i 's/ghcr.io\/koush\/scrypted/ghcr.io\/koush\/scrypted:nvidia/g' $DOCKER_COMPOSE_YML
+    fi
+fi
+
 readyn "Install avahi-daemon? This is the recommended for reliable HomeKit discovery and pairing."
 if [ "$yn" == "y" ]
 then
diff --git a/server/package-lock.json b/server/package-lock.json
index 452693706..e4d5593d4 100644
--- a/server/package-lock.json
+++ b/server/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@scrypted/server",
-  "version": "0.140.5",
+  "version": "0.140.6",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@scrypted/server",
-      "version": "0.140.5",
+      "version": "0.140.6",
       "hasInstallScript": true,
       "license": "ISC",
       "dependencies": {