Upgrade nvidia driver

Author: hgreebe

cookbooks/aws-parallelcluster-platform/attributes/platform.rb

@@ -17,7 +17,10 @@
 # NVidia
 default['cluster']['nvidia']['enabled'] = 'no'
 default['cluster']['nvidia']['driver_version'] = '570.86.15'
-default['cluster']['nvidia']['dcgm_version'] = '3.3.9'
+default['cluster']['nvidia']['dcgm_version'] = '3.3.6'
+if platform?('amazon') && node['platform_version'] == "2"
+  default['cluster']['nvidia']['driver_version'] = '550.127.08'
+end
 
 # DCV
 default['cluster']['dcv']['authenticator']['user'] = "dcvextauth"

cookbooks/aws-parallelcluster-platform/recipes/install/cuda.rb

@@ -23,9 +23,16 @@
 cuda_patch = '0'
 cuda_complete_version = "#{cuda_version}.#{cuda_patch}"
 cuda_version_suffix = '570.86.10'
-cuda_arch = arm_instance? ? 'linux_sbsa' : 'linux'
-cuda_url = "https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_#{cuda_complete_version}_#{cuda_version_suffix}_#{cuda_arch}.run"
 cuda_samples_version = '12.8'
+if platform?('amazon') && node['platform_version'] == "2"
+  cuda_version = '12.4'
+  cuda_patch = '1'
+  cuda_complete_version = "#{cuda_version}.#{cuda_patch}"
+  cuda_version_suffix = '550.54.15'
+  cuda_samples_version = '12.4'
+end
+cuda_arch = arm_instance? ? 'linux_sbsa' : 'linux'
+cuda_url = "https://developer.download.nvidia.com/compute/cuda/#{cuda_complete_version}/local_installers/cuda_#{cuda_complete_version}_#{cuda_version_suffix}_#{cuda_arch}.run"
 cuda_samples_url = "https://github.com/NVIDIA/cuda-samples/archive/refs/tags/v#{cuda_samples_version}.tar.gz"
 tmp_cuda_run = '/tmp/cuda.run'
 tmp_cuda_sample_archive = '/tmp/cuda-sample.tar.gz'