From ca1bc000df8ed1ae1f7c1d6b82cb71a5f22538ce Mon Sep 17 00:00:00 2001
From: Lightemerald <light_emerald@aostia.com>
Date: Fri, 21 Mar 2025 09:00:23 +0100
Subject: [PATCH] Adding nvidia container toolkit

---
 arch-setup.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/arch-setup.sh b/arch-setup.sh
index aa82a77..c35ef7c 100755
--- a/arch-setup.sh
+++ b/arch-setup.sh
@@ -426,6 +426,10 @@ install_packages_group() {
 		;;
 			"\"Containers\"")
 			install_packages containers_packages[@]
+            # if nvidia gpu is detected
+            if lspci | grep -Ei "nvidia" > /dev/null; then
+                sudo pacman -S --needed --noconfirm nvidia-container-toolkit
+            fi
 		;;
 			"\"U2F\"")
 			setup_u2f