File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
templates/al2023/provisioners Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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} .* "
202202sudo 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
205205if 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} .* "
209209else
210210 sudo dnf -y install nvidia-container-toolkit
211211fi
You can’t perform that action at this time.
0 commit comments