Upgrade NVIDIA driver and cuda version

Author: hgreebe

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

@@ -16,8 +16,11 @@
 
 # NVidia
 default['cluster']['nvidia']['enabled'] = 'no'
-default['cluster']['nvidia']['driver_version'] = '550.127.08'
+default['cluster']['nvidia']['driver_version'] = '570.86.15'
 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

@@ -19,13 +19,20 @@
 
 # Cuda installer from https://developer.nvidia.com/cuda-toolkit-archive
 # Cuda installer naming: cuda_11.8.0_520.61.05_linux
-cuda_version = '12.4'
-cuda_patch = '1'
+cuda_version = '12.8'
+cuda_patch = '0'
 cuda_complete_version = "#{cuda_version}.#{cuda_patch}"
-cuda_version_suffix = '550.54.15'
+cuda_version_suffix = '570.86.10'
+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 = "#{node['cluster']['artifacts_s3_url']}/dependencies/cuda/cuda_#{cuda_complete_version}_#{cuda_version_suffix}_#{cuda_arch}.run"
-cuda_samples_version = '12.4'
 cuda_samples_url = "#{node['cluster']['artifacts_s3_url']}/dependencies/cuda/samples/v#{cuda_samples_version}.tar.gz"
 tmp_cuda_run = '/tmp/cuda.run'
 tmp_cuda_sample_archive = '/tmp/cuda-sample.tar.gz'