Add kasm-gpu.sh

kasm-gpu.sh

@@ -0,0 +1,42 @@
+#!/bin/bash
+
+# Check for NVIDIA cards
+if ! lspci | grep -i nvidia > /dev/null; then
+    echo "No NVIDIA GPU detected"
+    exit 0
+fi
+
+add-apt-repository -y ppa:graphics-drivers/ppa
+
+curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
+  && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
+    sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
+    sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
+
+apt update
+apt install -y ubuntu-drivers-common
+
+# Run ubuntu-drivers and capture the output
+DRIVER_OUTPUT=$(ubuntu-drivers list 2>/dev/null)
+# Extract server driver versions using grep and regex
+# Pattern looks for nvidia-driver-XXX-server
+SERVER_VERSIONS=$(echo "$DRIVER_OUTPUT" | grep -o 'nvidia-driver-[0-9]\+-server' | grep -o '[0-9]\+' | sort -n)
+# Check if any server versions were found
+if [ -z "$SERVER_VERSIONS" ]; then
+    echo "Error: No NVIDIA server driver versions found." >&2
+    exit 1
+fi
+# Find the highest version number
+LATEST_VERSION=$(echo "$SERVER_VERSIONS" | tail -n 1)
+# Validate that the version is numeric
+if ! [[ "$LATEST_VERSION" =~ ^[0-9]+$ ]]; then
+    echo "Error: Invalid version number: $LATEST_VERSION" >&2
+    exit 2
+fi
+# Output only the version number
+echo "Latest version is: $LATEST_VERSION"
+ubuntu-drivers install "nvidia:$LATEST_VERSION-server"
+apt install -y "nvidia-utils-$LATEST_VERSION-server"
+# Install NVIDIA toolkit + configure for docker
+apt-get install -y nvidia-container-toolkit
+nvidia-ctk runtime configure --runtime=docker