docker: add flag for enabling/disabling nvidia-container-toolkit

hosts/desktop/configuration.nix

@@ -14,7 +14,13 @@ args@{
     ../../modules/nixos/blender.nix
     ../../modules/nixos/browsers.nix
     ../../modules/nixos/devenv.nix
-    (import ../../modules/nixos/docker.nix (args // { user = "pomp"; }))
+    (import ../../modules/nixos/docker.nix (
+      args
+      // {
+        user = "pomp";
+        enableNvidia = true;
+      }
+    ))
     ../../modules/nixos/fonts.nix
     (import ../../modules/nixos/gamemode.nix (args // { user = "pomp"; }))
     ../../modules/nixos/gnome-remote-desktop.nix

hosts/homelab-ryan/configuration.nix

@@ -3,7 +3,13 @@ args@{ pkgs, inputs, ... }:
 {
   imports = [
     ../../modules/nixos/devenv.nix
-    (import ../../modules/nixos/docker.nix (args // { user = "main"; }))
+    (import ../../modules/nixos/docker.nix (
+      args
+      // {
+        user = "main";
+        enableNvidia = false;
+      }
+    ))
     (import ../../modules/nixos/nix.nix (args // { user = "main"; }))
     ../../modules/nixos/nixvim.nix
     (import ../../modules/nixos/shell.nix (args // { user = "main"; }))

hosts/homelab2/configuration.nix

@@ -9,7 +9,13 @@ args@{
     ./hardware-configuration.nix
     ./modules/ssh.nix
     ../../modules/nixos/devenv.nix
-    (import ../../modules/nixos/docker.nix (args // { user = "homelab2"; }))
+    (import ../../modules/nixos/docker.nix (
+      args
+      // {
+        user = "homelab2";
+        enableNvidia = false;
+      }
+    ))
     ../../modules/nixos/fonts.nix
     ../../modules/nixos/gnome-remote-desktop.nix
     ../../modules/nixos/locale.nix

modules/nixos/docker.nix

@@ -1,4 +1,9 @@
-{ pkgs, user, ... }:
+{
+  pkgs,
+  user,
+  enableNvidia,
+  ...
+}:
 
 {
   users.users.${user}.extraGroups = [ "docker" ];
@@ -26,5 +31,5 @@
       features.cdi = true;
     };
   };
-  hardware.nvidia-container-toolkit.enable = true;
+  hardware.nvidia-container-toolkit.enable = enableNvidia;
 }