diff --git a/INSTALL/cuda-10-2.sh b/INSTALL/cuda-10-2.sh
index 4f9e7c34..2fb4144f 100644
--- a/INSTALL/cuda-10-2.sh
+++ b/INSTALL/cuda-10-2.sh
@@ -12,8 +12,9 @@ if [ -x "$(command -v apt)" ]; then
 
     sudo apt-get update -y
 
-    sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda-toolkit-10-2 -y --no-install-recommends
+    sudo apt-get -o Dpkg::Options::="--force-overwrite" install  cuda-toolkit-10-2 -y --no-install-recommends
     sudo apt-get -o Dpkg::Options::="--force-overwrite" install --fix-broken -y
+    sudo apt install nvidia-utils-440 nvidia-headless-440 -y
 
     # Install CUDA DNN
     wget https://cdn.shinobi.video/installers/libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb -O cuda-dnn.deb
diff --git a/INSTALL/cuda-10.sh b/INSTALL/cuda-10.sh
index d032cd1b..ac17cc22 100644
--- a/INSTALL/cuda-10.sh
+++ b/INSTALL/cuda-10.sh
@@ -12,6 +12,7 @@ if [ -x "$(command -v apt)" ]; then
 
     sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda-toolkit-10-0 -y --no-install-recommends
     sudo apt-get -o Dpkg::Options::="--force-overwrite" install --fix-broken -y
+    sudo apt install nvidia-utils-440 nvidia-headless-440 -y
 
     # Install CUDA DNN
     wget https://cdn.shinobi.video/installers/libcudnn7_7.6.5.32-1+cuda10.0_amd64.deb -O cuda-dnn.deb
diff --git a/INSTALL/cuda.sh b/INSTALL/cuda.sh
index 4f9e7c34..06c18129 100644
--- a/INSTALL/cuda.sh
+++ b/INSTALL/cuda.sh
@@ -12,8 +12,9 @@ if [ -x "$(command -v apt)" ]; then
 
     sudo apt-get update -y
 
-    sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda-toolkit-10-2 -y --no-install-recommends
+    sudo apt-get -o Dpkg::Options::="--force-overwrite" install nvidia-430 cuda-toolkit-10-2 -y --no-install-recommends
     sudo apt-get -o Dpkg::Options::="--force-overwrite" install --fix-broken -y
+    sudo apt install nvidia-utils-440 nvidia-headless-440 -y
 
     # Install CUDA DNN
     wget https://cdn.shinobi.video/installers/libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb -O cuda-dnn.deb