Merge pull request #373 from etiennedub/nvidia-open

Set Nvidia driver stream to proprietary version

data/common.yaml

@@ -294,7 +294,6 @@ profile::jupyterhub::hub::register_url: "https://mokey.%{lookup('terraform.data.
 profile::jupyterhub::hub::reset_pw_url: "https://mokey.%{lookup('terraform.data.domain_name')}/auth/forgotpw"
 
 profile::gpu::install::passthrough::packages:
-  - kmod-nvidia-latest-dkms # require to be first package, otherwise kmod-nivida is installed
   - nvidia-driver-cuda-libs
   - nvidia-driver
   - nvidia-driver-devel

site/profile/manifests/gpu.pp

@@ -82,6 +82,7 @@
 
 class profile::gpu::install::passthrough (
   Array[String] $packages,
+  String $nvidia_driver_stream = '555-dkms'
 ) {
   $os = "rhel${::facts['os']['release']['major']}"
   $arch = $::facts['os']['architecture']
@@ -92,6 +93,16 @@
     path    => ['/usr/bin'],
   }
 
+  package { 'nvidia-stream':
+    ensure      => present,
+    name        => "nvidia-driver:${nvidia_driver_stream}",
+    provider    => dnfmodule,
+    enable_only => true,
+    require     => [
+      Exec['cuda-repo'],
+    ]
+  }
+
   $mig_profile = lookup("terraform.instances.${facts['networking']['hostname']}.specs.mig", Variant[Undef, Hash[String, Integer]], undef, {})
   class { 'profile::gpu::config::mig':
     mig_profile => $mig_profile,
@@ -101,6 +112,7 @@
   package { $packages:
     ensure  => 'installed',
     require => [
+      Package['nvidia-stream'],
       Package['kernel-devel'],
       Exec['cuda-repo'],
       Yumrepo['epel'],