Isolated partition install nvidia-persistenced (#2081)

Co-authored-by: Wyatt Hicken <whois@amazon.com>
awslabs · Dec 4, 2024 · da3591e · da3591e
1 parent 5b25e4d
commit da3591e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/templates/al2023/provisioners/install-nvidia-driver.sh b/templates/al2023/provisioners/install-nvidia-driver.sh
@@ -136,9 +136,10 @@ sudo systemctl enable set-nvidia-clocks.service
 ################################################################################
 sudo dnf -y install nvidia-fabric-manager
 
-# NVIDIA Container toolkit needs to be locally installed for isolated partitions
+# NVIDIA Container toolkit needs to be locally installed for isolated partitions, also install NVIDIA-Persistenced
 if is-isolated-partition; then
   install-nvidia-container-toolkit
+  sudo dnf -y install "nvidia-persistenced-${NVIDIA_DRIVER_MAJOR_VERSION}.*"
 else
   sudo dnf -y install nvidia-container-toolkit
 fi