Add darwin configuration into

modules/system/nixos/graphics/cuda.nix

@@ -0,0 +1,36 @@
+{
+  pkgs,
+  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;
+    environment.systemPackages = with pkgs.cudaPackages; [
+      cudatoolkit
+      cudnn
+      libcublas
+      libcufile
+      libcusparse
+      nccl
+    ];
+
+    environment.variables = {
+      CUDA_PATH = "${pkgs.cudatoolkit}";
+      LD_LIBRARY_PATH = "${pkgs.linuxPackages.nvidia_x11}/lib:$LD_LIBRARY_PATH";
+      EXTRA_LDFLAGS = "-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib";
+      EXTRA_CCFLAGS = "-I/usr/include";
+    };
+
+  };
+
+}

modules/system/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;
+    };
+
+  };
+}