Enable NVIDIA Plugin automatically for AL2023 GPU AMIs (#8199)

Signed-off-by: Davanum Srinivas <[email protected]>
eksctl-io · Feb 6, 2025 · 987f978 · 987f978
1 parent 731b521
commit 987f978
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pkg/eks/tasks.go b/pkg/eks/tasks.go
@@ -339,19 +339,21 @@ func (c *ClusterProvider) ClusterTasksForNodeGroups(cfg *api.ClusterConfig, inst
 	for _, ng := range cfg.NodeGroups {
 		clusterRequiresNeuronDevicePlugin = clusterRequiresNeuronDevicePlugin ||
 			api.HasInstanceType(ng, instanceutils.IsNeuronInstanceType)
-		// Only AL2 requires the NVIDIA device plugin
+		// Only AL2/AL2023 requires the NVIDIA device plugin
 		clusterRequiresNvidiaDevicePlugin = clusterRequiresNvidiaDevicePlugin ||
 			(api.HasInstanceType(ng, instanceutils.IsNvidiaInstanceType) &&
-				ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2)
+				ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2 ||
+				ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2023)
 		efaEnabled = efaEnabled || api.IsEnabled(ng.EFAEnabled)
 	}
 	for _, ng := range cfg.ManagedNodeGroups {
 		clusterRequiresNeuronDevicePlugin = clusterRequiresNeuronDevicePlugin ||
 			api.HasInstanceTypeManaged(ng, instanceutils.IsNeuronInstanceType)
-		// Only AL2 requires the NVIDIA device plugin
+		// Only AL2/AL2023 requires the NVIDIA device plugin
 		clusterRequiresNvidiaDevicePlugin = clusterRequiresNvidiaDevicePlugin ||
 			(api.HasInstanceTypeManaged(ng, instanceutils.IsNvidiaInstanceType) &&
-				ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2)
+				ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2 ||
+				ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2023)
 		efaEnabled = efaEnabled || api.IsEnabled(ng.EFAEnabled)
 	}
 	if clusterRequiresNeuronDevicePlugin {