nixos/adrastea: initialize NVIDIA GPU on startup, in systemd service

inclyc · Jun 11, 2024 · 3bd8538 · 3bd8538
1 parent 43a8c8d
commit 3bd8538
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/nixos/configurations/adrastea/nvidia.nix b/nixos/configurations/adrastea/nvidia.nix
@@ -14,4 +14,15 @@ in
   environment.sessionVariables = {
     NVIDIA_KERNEL = nvidia;
   };
+
+  # Initialize nvidia gpu card using "nvidia-smi" in root.
+  systemd.services."nvidia-init" = {
+    description = "Initialize NVIDIA GPU";
+    serviceConfig = {
+      Type = "oneshot";
+      ExecStart = "${nvidia.bin}/bin/nvidia-smi";
+    };
+
+    wantedBy = [ "multi-user.target" ];
+  };
 }