chore(al2023): update kernel package names (#2345)

Author: ndbaker1

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

@@ -29,9 +29,14 @@ metadata_expire=0" | sudo tee /etc/yum.repos.d/neuron.repo
 ################################################################################
 ### Install packages ###########################################################
 ################################################################################
-
+KERNEL_PACKAGE="kernel"
+if [[ "$(uname -r)" == 6.12.* ]]; then
+  KERNEL_PACKAGE="kernel6.12"
+fi
 sudo dnf -y install \
-  kernel-devel-$(uname -r) \
-  kernel-headers-$(uname -r)
+  "${KERNEL_PACKAGE}-devel" \
+  "${KERNEL_PACKAGE}-headers"
+
+sudo dnf versionlock 'kernel*'
 
 sudo dnf install -y aws-neuronx-dkms aws-neuronx-tools

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

@@ -70,16 +70,17 @@ sudo mv ${WORKING_DIR}/gpu/kmod-util /usr/bin/
 sudo mkdir -p /etc/dkms
 echo "MAKE[0]=\"'make' -j$(grep -c processor /proc/cpuinfo) module\"" | sudo tee /etc/dkms/nvidia.conf
 
+KERNEL_PACKAGE="kernel"
 if [[ "$(uname -r)" == 6.12.* ]]; then
-  sudo dnf -y install kernel6.12-modules-extra-$(uname -r)
-else
-  sudo dnf -y install kernel-modules-extra-$(uname -r)
+  KERNEL_PACKAGE="kernel6.12"
 fi
-
 sudo dnf -y install \
-  kernel-devel-$(uname -r) \
-  kernel-headers-$(uname -r) \
-  kernel-modules-extra-common-$(uname -r)
+  "${KERNEL_PACKAGE}-devel" \
+  "${KERNEL_PACKAGE}-headers" \
+  "${KERNEL_PACKAGE}-modules-extra" \
+  "${KERNEL_PACKAGE}-modules-extra-common"
+
+sudo dnf versionlock 'kernel*'
 
 # Install dkms dependency from amazonlinux repo
 sudo dnf -y install patch

templates/al2023/provisioners/install-worker.sh

@@ -66,6 +66,19 @@ sudo dnf install -y \
   pigz \
   python3-dnf-plugin-versionlock
 
+# we need to handle different kernel packages depending on the namespace
+# associated with the minor version.
+KERNEL_PACKAGE="kernel"
+if [[ "$(uname -r)" == 6.12.* ]]; then
+  KERNEL_PACKAGE="kernel6.12"
+fi
+sudo dnf -y install \
+  "${KERNEL_PACKAGE}-devel" \
+  "${KERNEL_PACKAGE}-headers"
+
+# versionlock kernel packages so they remain consistent.
+sudo dnf versionlock 'kernel*'
+
 ################################################################################
 ### Networking #################################################################
 ################################################################################