Change in folders organisation

modules/system/options/nixos/graphics/cuda.nix

@@ -0,0 +1,15 @@
+{ lib, config, ... }:
+with lib;
+let
+  cfg = config.os.virtualisation.cuda;
+in
+{
+  options.os.virtualisation.cuda = {
+    enable = mkEnableOption "System CUDA service";
+  };
+  config = mkIf cfg.enable {
+
+    hardware.nvidia-container-toolkit.enable = true;
+
+  };
+}

modules/system/options/nixos/graphics/nvidia.nix

@@ -0,0 +1,24 @@
+{ lib, config, ... }:
+with lib;
+let
+  cfg = config.os.graphics.nvidia;
+in
+{
+  options.os.graphics.nvidia = {
+    enable = mkEnableOption "System nvidia service";
+  };
+  config = mkIf cfg.enable {
+
+    services.xserver.enable = true;
+    services.xserver.videoDrivers = [ "nvidia" ];
+
+    hardware.graphics.enable = true;
+    hardware.nvidia = {
+      open = true;
+      nvidiaSettings = false;
+      modesetting.enable = true; # Wayland requirements
+      powerManagement.enable = true;
+    };
+
+  };
+}