diff --git a/INSTALL/cuda-10-2.sh b/INSTALL/cuda-10-2.sh
index 2fb4144f..364be7ea 100644
--- a/INSTALL/cuda-10-2.sh
+++ b/INSTALL/cuda-10-2.sh
@@ -29,7 +29,7 @@ fi
 if [ -x "$(command -v yum)" ]; then
     sudo yum-config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
     sudo yum clean all
-    sudo yum -y install nvidia-driver-latest-dkms cuda
+    sudo yum -y install nvidia-driver-latest-dkms cuda-toolkit-10-2
     sudo yum -y install cuda-drivers
     wget https://cdn.shinobi.video/installers/libcudnn7-7.6.5.33-1.cuda10.2.x86_64.rpm -O cuda-dnn.rpm
     sudo yum -y localinstall cuda-dnn.rpm
diff --git a/INSTALL/cuda-10.sh b/INSTALL/cuda-10.sh
index 192265fb..2b5f7a04 100644
--- a/INSTALL/cuda-10.sh
+++ b/INSTALL/cuda-10.sh
@@ -29,7 +29,7 @@ if [ -x "$(command -v yum)" ]; then
     wget https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-10.0.130-1.x86_64.rpm
     sudo rpm -i cuda-repo-rhel7-10.0.130-1.x86_64.rpm
     sudo yum clean all
-    sudo yum install cuda
+    sudo yum install cuda-toolkit-10-0 -y
     wget https://cdn.shinobi.video/installers/libcudnn7-7.6.5.32-1.cuda10.0.x86_64.rpm -O cuda-dnn.rpm
     sudo yum -y localinstall cuda-dnn.rpm
     wget https://cdn.shinobi.video/installers/libcudnn7-devel-7.6.5.32-1.cuda10.0.x86_64.rpm -O cuda-dnn-dev.rpm