1{
 2  config,
 3  lib,
 4  nixos-hardware,
 5  ...
 6}: {
 7  imports = [
 8    nixos-hardware.nixosModules.common-gpu-nvidia
 9  ];
10
11  services.xserver.videoDrivers = [
12    "nvidia"
13    "modesetting" # needed for offload mode
14  ];
15  boot.initrd.kernelModules = ["nvidia"];
16
17  # Nvidia ada lovelace
18  hardware.nvidia = {
19    package = config.boot.kernelPackages.nvidiaPackages.stable;
20    open = true;
21    prime = {
22      intelBusId = "PCI:0:2:0";
23      nvidiaBusId = "PCI:1:0:0";
24    };
25    primeBatterySaverSpecialisation = true;
26    powerManagement = {
27      enable = true;
28      finegrained = true;
29    };
30  };
31
32  hardware.nvidia-container-toolkit.enable = true;
33  hardware.graphics = {
34    enable = true;
35    # # needed by nvidia-docker
36    # enable32Bit = true;
37  };
38
39  nixpkgs.config.cudaSupport = true;
40}