diff --git a/install.sh b/install.sh
index 789bb38..1f4dbbe 100644
--- a/install.sh
+++ b/install.sh
@@ -69,7 +69,9 @@ gpu=$(lspci | grep -i 'nvidia')
 
 if [[ -n $gpu ]]; then
   printf 'Une carte Nvidia a été détectée: %s\n' "$gpu"
-  apt install nvidia-smi
+  apt install linux-headers-amd64
+  apt install firmware-misc-nonfree nvidia-driver nvidia-smi
+  apt install nvidia-cuda-dev nvidia-cuda-toolkit
 else
   printf 'Aucune carte Nvidia détectée\n'
 fi