diff --git a/cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/nvidia_driver_alinux2023.rb b/cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/nvidia_driver_alinux2023.rb
index 81f6261aa9..74b049ce04 100644
--- a/cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/nvidia_driver_alinux2023.rb
+++ b/cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/nvidia_driver_alinux2023.rb
@@ -33,5 +33,6 @@ def compiler_path
 end
 
 def extra_packages
-  %w(kernel-modules-extra kernel-modules-extra-common)
+  kernel_version = node['kernel']['release'].chomp('.x86_64').chomp('.aarch64')
+  ["kernel-modules-extra-#{kernel_version}", "kernel-modules-extra-common-#{kernel_version}"]
 end