moved

hosts/modules/nvidia.nix

@@ -0,0 +1,35 @@
+{
+  config,
+  pkgs,
+  ...
+}: {
+  services.xserver.videoDrivers = ["nvidia"];
+
+  hardware.nvidia = {
+    package = config.boot.kernelPackages.nvidiaPackages.stable;
+    open = true;
+    modesetting.enable = true; # must be true for Wayland
+    powerManagement.enable = false;
+    nvidiaSettings = true;
+  };
+
+  hardware.graphics = {
+    enable = true;
+    enable32Bit = true;
+  };
+
+  programs.nix-ld = {
+    enable = true;
+    libraries = with pkgs; [
+      stdenv.cc.cc.lib
+      cudaPackages.cudatoolkit.lib
+      cudaPackages.cudnn
+      cudaPackages.nccl
+    ];
+  };
+
+  environment.systemPackages = with pkgs; [
+    glxinfo
+    cudaPackages.cudatoolkit
+  ];
+}