Skip to content

Commit

Permalink
Merge pull request #383 from etiennedub/fix_nvidia_service
Browse files Browse the repository at this point in the history
Use dynamic user for nvidia-persistenced service
  • Loading branch information
cmd-ntrf authored Sep 26, 2024
2 parents c1aae73 + 6db5ba5 commit 17e0b96
Showing 1 changed file with 4 additions and 15 deletions.
19 changes: 4 additions & 15 deletions site/profile/manifests/gpu.pp
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@

class profile::gpu::install::passthrough (
Array[String] $packages,
String $nvidia_driver_stream = '555-dkms'
String $nvidia_driver_stream = '560-dkms'
) {
$os = "rhel${::facts['os']['release']['major']}"
$arch = $::facts['os']['architecture']
Expand Down Expand Up @@ -122,26 +122,15 @@
# Used by slurm-job-exporter to export GPU metrics
-> package { 'datacenter-gpu-manager': }

-> file { '/run/nvidia-persistenced':
ensure => directory,
owner => 'nvidia-persistenced',
group => 'nvidia-persistenced',
mode => '0755',
}

-> augeas { 'nvidia-persistenced.service':
context => '/files/lib/systemd/system/nvidia-persistenced.service/Service',
changes => [
'set User/value nvidia-persistenced',
'set Group/value nvidia-persistenced',
'set DynamicUser/value yes',
'set StateDirectory/value nvidia-persistenced',
'set RuntimeDirectory/value nvidia-persistenced',
'rm ExecStart/arguments',
],
}

file { '/usr/lib/tmpfiles.d/nvidia-persistenced.conf':
content => 'd /run/nvidia-persistenced 0755 nvidia-persistenced nvidia-persistenced -',
mode => '0644',
}
}

class profile::gpu::config::mig (
Expand Down

0 comments on commit 17e0b96

Please sign in to comment.