test upgrade nvidia driver version

Author: hgreebe

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

@@ -16,8 +16,8 @@
 
 # NVidia
 default['cluster']['nvidia']['enabled'] = 'no'
-default['cluster']['nvidia']['driver_version'] = '550.127.08'
-default['cluster']['nvidia']['dcgm_version'] = '3.3.6'
+default['cluster']['nvidia']['driver_version'] = '570.86.15'
+default['cluster']['nvidia']['dcgm_version'] = '3.3.9'
 
 # DCV
 default['cluster']['dcv']['authenticator']['user'] = "dcvextauth"

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

@@ -19,14 +19,14 @@
 
 # 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_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"
+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'
+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'
 

cookbooks/aws-parallelcluster-platform/resources/fabric_manager/partial/_fabric_manager_install_debian.rb

@@ -38,6 +38,10 @@ def arch_suffix
   arm_instance? ? 'arm64' : 'amd64'
 end
 
+def arch_name
+  arm_instance? ? 'sbsa' : 'x86_64'
+end
+
 def fabric_manager_url
-  "#{node['cluster']['artifacts_s3_url']}/dependencies/nvidia_fabric/#{platform}/#{fabric_manager_package}_#{fabric_manager_version}-1_#{arch_suffix}.deb"
+  "https://developer.download.nvidia.com/compute/cuda/repos/#{platform}/#{arch_name}/#{fabric_manager_package}_#{fabric_manager_version}-1_#{arch_suffix}.deb"
 end

cookbooks/aws-parallelcluster-platform/resources/fabric_manager/partial/_fabric_manager_install_rhel.rb

@@ -39,6 +39,10 @@ def arch_suffix
   arm_instance? ? 'aarch64' : 'x86_64'
 end
 
+def arch_name
+  arm_instance? ? 'sbsa' : 'x86_64'
+end
+
 def fabric_manager_url
-  "#{node['cluster']['artifacts_s3_url']}/dependencies/nvidia_fabric/#{platform}/#{fabric_manager_package}-#{fabric_manager_version}-1.#{arch_suffix}.rpm"
+  "https://developer.download.nvidia.com/compute/cuda/repos/#{platform}/#{arch_name}/#{fabric_manager_package}-#{fabric_manager_version}-1.#{arch_suffix}.rpm"
 end

cookbooks/aws-parallelcluster-platform/resources/nvidia_dcgm/partial/_nvidia_dcgm_debian.rb

@@ -33,8 +33,12 @@
   end
 end
 
+def arch_name
+  arm_instance? ? 'sbsa' : 'x86_64'
+end
+
 def dcgm_url
-  "#{node['cluster']['artifacts_s3_url']}/dependencies/nvidia_dcgm/#{platform}/#{dcgm_package}_#{package_version}_#{arch_suffix}.deb"
+  "https://developer.download.nvidia.com/compute/cuda/repos/#{platform}/#{arch_name}/#{dcgm_package}_#{package_version}_#{arch_suffix}.deb"
 end
 
 def dcgm_package

cookbooks/aws-parallelcluster-platform/resources/nvidia_dcgm/partial/_nvidia_dcgm_rhel.rb

@@ -33,8 +33,12 @@
   end
 end
 
+def arch_name
+  arm_instance? ? 'sbsa' : 'x86_64'
+end
+
 def dcgm_url
-  "#{node['cluster']['artifacts_s3_url']}/dependencies/nvidia_dcgm/#{platform}/#{dcgm_package}-#{package_version}-1-#{arch_suffix}.rpm"
+  "https://developer.download.nvidia.com/compute/cuda/repos/#{platform}/#{arch_name}/#{dcgm_package}-#{package_version}-1-#{arch_suffix}.rpm"
 end
 
 def dcgm_package

cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/partial/_nvidia_driver_common.rb

@@ -96,7 +96,7 @@ def _nvidia_driver_version
 end
 
 def nvidia_driver_url
-  "#{node['cluster']['artifacts_s3_url']}/dependencies/nvidia_driver/NVIDIA-Linux-#{nvidia_arch}-#{_nvidia_driver_version}.run"
+  "https://us.download.nvidia.com/tesla/#{_nvidia_driver_version}/NVIDIA-Linux-#{nvidia_arch}-#{_nvidia_driver_version}.run"
 end
 
 def nvidia_driver_enabled?