Cherry pick nvidia (#7)

* Set Nvidia driver stream to proprietary version

* Switch to version 555

* Set Nvidia driver stream to proprietary version

* squid cache at localscratch root

---------

Co-authored-by: Etienne <etienne.dubeau@calculquebec.ca>
Co-authored-by: Félix-Antoine Fortin <felix-antoine.fortin@calculquebec.ca>
Co-authored-by: poquirion <pierre-olivier.quirion@calculquebec.ca>

data/common.yaml

@@ -290,3 +290,12 @@ profile::reverse_proxy::subdomains:
 
 profile::jupyterhub::hub::register_url: "https://mokey.%{lookup('terraform.data.domain_name')}/auth/signup"
 profile::jupyterhub::hub::reset_pw_url: "https://mokey.%{lookup('terraform.data.domain_name')}/auth/forgotpw"
+profile::gpu::install::passthrough::packages:
+  - nvidia-driver-cuda-libs
+  - nvidia-driver
+  - nvidia-driver-devel
+  - nvidia-driver-libs
+  - nvidia-driver-NVML
+  - nvidia-modprobe
+  - nvidia-xconfig
+  - nvidia-persistenced

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']
@@ -97,6 +98,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,
@@ -106,6 +117,7 @@
   package { $packages:
     ensure  => 'installed',
     require => [
+      Package['nvidia-stream'],
       Package['kernel-devel'],
       Exec['cuda-repo'],
       Yumrepo['epel'],

site/profile/manifests/squid.pp

@@ -52,7 +52,7 @@
   squid::http_access { 'CLUSTER_NETWORK CVMFS':
     action => 'allow',
   }
-  squid::cache_dir { '/localscratch/var/spool/squid':
+  squid::cache_dir { '/localscratch/squid':
     type    => 'ufs',
     options => "${cache_size} 16 256",
   }