diff --git a/bootstrap.sh b/bootstrap.sh
index ac7e2b7..c7657ed 100755
--- a/bootstrap.sh
+++ b/bootstrap.sh
@@ -5,7 +5,7 @@ locale-gen en_US.UTF-8
 # Set the hostname
 echo "scarif.space" > /etc/hostname
 
-source /opt/ai-mouse/.env
+source /opt/mouse/.env
 
 echo "------- Generating system users -------"
 useradd -m -p $(echo $USER_PASSWORD | openssl passwd -1 -stdin) chris
@@ -20,7 +20,7 @@ echo "------- Setting up SSH -------"
 mkdir -p /home/chris/.ssh
 cp /root/.ssh/authorized_keys /home/chris/.ssh/authorized_keys
 rm /etc/ssh/sshd_config
-ln -s /opt/ai-mouse/sshd_config /etc/ssh/sshd_config
+ln -s /opt/mouse/sshd_config /etc/ssh/sshd_config
 echo "Welcome to The Mouse!" > /etc/motd
 systemctl restart sshd
 
@@ -30,13 +30,12 @@ ufw default allow outgoing
 ufw allow 629
 ufw --force enable
 
-#echo "------- Install Cuda -------"
-#apt-key del 7fa2af80
-#wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu24-4/x86_64/cuda-keyring_1.1-1_all.deb
-#sudo dpkg -i cuda-keyring_1.1-1_all.deb
-#sudo apt-get update
-#sudo apt-get install cuda-toolkit
-#sudo apt-get install nvidia-gds
+echo "------- Install Cuda -------"
+apt remove -y vultr-nvidia-client-drivers
+wget -q -O /tmp/vultr_nvidia_driver.zip "http://169.254.169.254/latest/nvidia_linux_driver_installer_url"
+unzip -qq -o /tmp/vultr_nvidia_driver.zip -d /
+bash /opt/nvidia/install.sh
+rm -f /tmp/vultr_nvidia_driver.zip
 
 # echo "------- Starting docker -------"
 # systemctl enable docker --now