Add --no-drm flag to nvidia installation for ubuntu24

Author: hgreebe

cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/nvidia_driver_ubuntu22+.rb

@@ -22,6 +22,12 @@ def rebuild_initramfs?
   true
 end
 
+def drm
+  if node['platform_version'].to_i == 24
+    '--no-drm'
+  end
+end
+
 def set_compiler?
   true
 end

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

@@ -79,7 +79,7 @@
     cwd '/tmp'
     code <<-NVIDIA
       set -e
-      #{compiler_path} ./nvidia.run --silent --dkms --disable-nouveau -m=#{nvidia_kernel_module}
+      #{compiler_path} ./nvidia.run --silent --dkms #{drm}--disable-nouveau -m=#{nvidia_kernel_module}
       rm -f /tmp/nvidia.run
     NVIDIA
     creates '/usr/bin/nvidia-smi'
@@ -91,6 +91,10 @@
   end if rebuild_initramfs?
 end
 
+def drm
+  ''
+end
+
 def _nvidia_driver_version
   nvidia_driver_version || node['cluster']['nvidia']['driver_version']
 end

cookbooks/aws-parallelcluster-platform/spec/unit/resources/nvidia_driver_spec.rb

@@ -249,6 +249,8 @@ def self.setup(chef_run, nvidia_driver_version: nil)
             end
           end
 
+          drm = platform == 'ubuntu' && version == '24' ? '--no-drm' : ''
+
           it 'installs nvidia driver' do
             is_expected.to run_bash('nvidia.run advanced')
               .with(
@@ -257,7 +259,7 @@ def self.setup(chef_run, nvidia_driver_version: nil)
                 cwd: '/tmp',
                 creates: '/usr/bin/nvidia-smi'
               )
-              .with_code(%r{CC=/usr/bin/gcc10-gcc ./nvidia.run --silent --dkms --disable-nouveau -m=#{kernel_module}})
+              .with_code(%r{CC=/usr/bin/gcc10-gcc ./nvidia.run --silent --dkms #{drm} --disable-nouveau -m=#{kernel_module}})
               .with_code(%r{rm -f /tmp/nvidia.run})
           end
         elsif platform == 'ubuntu' && version == '22.04'