From 262235bf434331e68790732193488b71671d0384 Mon Sep 17 00:00:00 2001
From: Moe <github@m03.ca>
Date: Sat, 26 Nov 2022 16:58:49 -0800
Subject: [PATCH] Update CUDA installer

---
 INSTALL/cuda.sh | 55 +++++++++++++++++++++----------------------------
 1 file changed, 23 insertions(+), 32 deletions(-)

diff --git a/INSTALL/cuda.sh b/INSTALL/cuda.sh
index 06c18129..fee745ba 100644
--- a/INSTALL/cuda.sh
+++ b/INSTALL/cuda.sh
@@ -4,47 +4,38 @@ echo "-- Installing CUDA Toolkit and CUDA DNN --"
 echo "------------------------------------------"
 # Install CUDA Drivers and Toolkit
 if [ -x "$(command -v apt)" ]; then
-    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
-    sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
-    sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
-    sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
-    sudo apt-get update
-
+    echo "Installing cuda-toolkit-10-0"
+    wget https://cdn.shinobi.video/installers/cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb -O cuda.deb
+    dpkg -i cuda.deb
+    sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
     sudo apt-get update -y
+    # CUDA Toolkit
+    sudo apt install cuda-toolkit-10-0
 
-    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
+    # Driver
+    echo "Installing nvidia-driver-515-server"
+    sudo apt install nvidia-driver-515-server
 
-    # Install CUDA DNN
-    wget https://cdn.shinobi.video/installers/libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb -O cuda-dnn.deb
+    # CUDNN
+    echo "Installing CUDNN"
+    wget https://cdn.shinobi.video/installers/libcudnn7_7.6.5.32-1+cuda10.0_amd64.deb -O cuda-dnn.deb
     sudo dpkg -i cuda-dnn.deb
-    wget https://cdn.shinobi.video/installers/libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb -O cuda-dnn-dev.deb
+    wget https://cdn.shinobi.video/installers/libcudnn7-dev_7.6.5.32-1+cuda10.0_amd64.deb -O cuda-dnn-dev.deb
     sudo dpkg -i cuda-dnn-dev.deb
     echo "-- Cleaning Up --"
     # Cleanup
+    sudo rm cuda.deb
     sudo rm cuda-dnn.deb
     sudo rm cuda-dnn-dev.deb
+    echo "------------------------------"
+    echo "Reboot is required. Do it now?"
+    echo "------------------------------"
+    echo "(y)es or (N)o. Default is No."
+    read rebootTheMachineHomie
+    if [ "$rebootTheMachineHomie" = "y" ] || [ "$rebootTheMachineHomie" = "Y" ]; then
+        sudo reboot
+    fi
 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 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
-    wget https://cdn.shinobi.video/installers/libcudnn7-devel-7.6.5.33-1.cuda10.2.x86_64.rpm -O cuda-dnn-dev.rpm
-    sudo yum -y localinstall cuda-dnn-dev.rpm
-    echo "-- Cleaning Up --"
-    sudo rm cuda-dnn.rpm
-    sudo rm cuda-dnn-dev.rpm
-fi
-
-echo "------------------------------"
-echo "Reboot is required. Do it now?"
-echo "------------------------------"
-echo "(y)es or (N)o. Default is No."
-read rebootTheMachineHomie
-if [ "$rebootTheMachineHomie" = "y" ] || [ "$rebootTheMachineHomie" = "Y" ]; then
-    sudo reboot
+    echo "CentOS no longer supported by this installer."
 fi