diff --git a/install/install_nomad.sh b/install/install_nomad.sh
index bf7dc15..1230fce 100644
--- a/install/install_nomad.sh
+++ b/install/install_nomad.sh
@@ -596,7 +596,7 @@ verify_gpu_setup() {
   echo -e "${YELLOW}===========================================${RESET}\\n"
   
   # Summary
-  if command -v nvidia-smi &> /dev/null && docker info 2>/dev/null; then
+  if command -v nvidia-smi &> /dev/null && docker info 2>/dev/null | grep "nvidia"; then
     echo -e "${GREEN}#${RESET} GPU acceleration is properly configured! The AI Assistant will use your GPU.\\n"
   else
     echo -e "${YELLOW}#${RESET} GPU acceleration not detected. The AI Assistant will run in CPU-only mode.\\n"