Skip to content

Commit 405b829

Browse files
whoixWyatt Hicken
andauthored
Pin NVIDIA Driver version for Isolated regions (#2472)
* Pin NVIDIA Driver version for Isolated regions * Isolated region specify cuda --------- Co-authored-by: Wyatt Hicken <whois@amazon.com>
1 parent 2c5698d commit 405b829

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -198,14 +198,14 @@ sudo systemctl enable set-nvidia-clocks.service
198198
################################################################################
199199
### Install other dependencies #################################################
200200
################################################################################
201-
sudo dnf -y install nvidia-fabric-manager
201+
sudo dnf -y install "nvidia-fabric-manager-${NVIDIA_DRIVER_MAJOR_VERSION}.*"
202202
sudo dnf -y install "nvidia-imex-${NVIDIA_DRIVER_MAJOR_VERSION}.*"
203203

204204
# NVIDIA Container toolkit needs to be locally installed for isolated partitions, also install NVIDIA-Persistenced
205205
if is-isolated-partition; then
206206
sudo dnf -y install nvidia-container-toolkit
207207
sudo dnf -y install "nvidia-persistenced-${NVIDIA_DRIVER_MAJOR_VERSION}.*"
208-
sudo dnf -y install "nvidia-driver"
208+
sudo dnf -y install "nvidia-driver-cuda-${NVIDIA_DRIVER_MAJOR_VERSION}.*"
209209
else
210210
sudo dnf -y install nvidia-container-toolkit
211211
fi

0 commit comments

Comments
 (0)