cleanup

Author: mselim00

templates/al2023/provisioners/install-nvidia-driver.sh

@@ -82,10 +82,14 @@ sudo dnf -y install \
   kernel-headers-$(uname -r) \
   kernel-modules-extra-common-$(uname -r)
 
-# Install dkms dependency from amazonlinux
+# Install dkms dependency from amazonlinux repo
 sudo dnf -y install patch
 # Install dkms from the cuda repo
+if is-isolated-partition; then
 sudo dnf -y --disablerepo="*" --enablerepo="cuda*" install dkms
+else
+  sudo dnf -y --disablerepo="*" --enablerepo="cuda*" install dkms
+fi
 
 function archive-open-kmods() {
   echo "Archiving open kmods"