Add darwin configuration into

2026-02-11 13:15:26 +01:00
parent 80e5d0b8b5
commit e75b09a750
71 changed files with 562 additions and 151 deletions
@@ -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";
+    };
+
+  };
+
+}