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