build(nvidia): install nvidia-imex driver (#2136)

awslabs · Feb 13, 2025 · 0359363 · 0359363
1 parent 0030ac5
commit 0359363
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/templates/al2023/provisioners/install-nvidia-driver.sh b/templates/al2023/provisioners/install-nvidia-driver.sh
@@ -119,6 +119,7 @@ sudo systemctl enable set-nvidia-clocks.service
 ### Install other dependencies #################################################
 ################################################################################
 sudo dnf -y install nvidia-fabric-manager
+sudo dnf -y install "nvidia-imex-${NVIDIA_DRIVER_MAJOR_VERSION}.*"
 
 # NVIDIA Container toolkit needs to be locally installed for isolated partitions, also install NVIDIA-Persistenced
 if is-isolated-partition; then