From 0269888da5b3d7483e0184b63920e4b6d88c9b2d Mon Sep 17 00:00:00 2001
From: Moe <github@m03.ca>
Date: Sat, 5 Oct 2024 12:22:23 -0700
Subject: [PATCH] Create cuda-12.sh

---
 INSTALL/cuda-12.sh | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 INSTALL/cuda-12.sh

diff --git a/INSTALL/cuda-12.sh b/INSTALL/cuda-12.sh
new file mode 100644
index 00000000..ef0ac003
--- /dev/null
+++ b/INSTALL/cuda-12.sh
@@ -0,0 +1,44 @@
+#!/bin/sh
+echo "------------------------------------------"
+echo "-- Installing CUDA Toolkit and CUDA DNN --"
+echo "------------------------------------------"
+# Install CUDA Drivers and Toolkit
+echo "Install CUDA Drivers and Toolkit?"
+echo "(y)es or (N)o"
+read installTheStuffHomie
+if [ "$installTheStuffHomie" = "y" ] || [ "$installTheStuffHomie" = "Y" ]; then
+    if [ -x "$(command -v apt)" ]; then
+        # CUDA Toolkit
+        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
+        sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
+        echo "Downloading CUDA Toolkit..."
+        wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-ubuntu2204-12-0-local_12.0.0-525.60.13-1_amd64.deb
+        sudo dpkg -i cuda-repo-ubuntu2204-12-0-local_12.0.0-525.60.13-1_amd64.deb
+        sudo cp /var/cuda-repo-ubuntu2204-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
+        sudo apt-get update
+        sudo apt-get -y install cuda-toolkit-12-0
+
+        # Driver
+        echo "Installing nvidia-driver-515-server"
+        sudo apt install nvidia-driver-515-server -y
+
+        # Install CUDA DNN
+        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/libcudnn8_8.8.1.3-1+cuda12.0_amd64.deb -O cuda-dnn.deb --no-verbose
+        sudo dpkg -i cuda-dnn.deb
+        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/libcudnn8-dev_8.8.1.3-1+cuda12.0_amd64.deb -O cuda-dnn-dev.deb --no-verbose
+        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
+    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
+    fi
+fi