modules/teh-nix/nvidia.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

{ ... }:
{
    flake.modules.nixos.teh-nix =
        { config, pkgs, lib, ... }:
        let
            nvidia-offload = pkgs.writeShellScriptBin "prime-run" ''
export __NV_PRIME_RENDER_OFFLOAD=1
export __NV_PRIME_RENDER_OFFLOAD_PROVIDER=NVIDIA-G0
export __GLX_VENDOR_LIBRARY_NAME=nvidia
export __VK_LAYER_NV_optimus=NVIDIA_only
exec "$@"
'';
        in
        {
            options.teh-nix.nvidia = with lib; {
                enable = mkEnableOption "Enable nvidia";
            };

            config = lib.mkIf config.teh-nix.nvidia.enable {
                nixpkgs.config.allowUnfree = true;
                hardware.graphics.enable = true;
                hardware.nvidia.modesetting.enable = true;
                # hardware.nvidia.dynamicBoost.enable = true;
                hardware.nvidia.open = true;
                hardware.nvidia.prime = {
                    # sync.enable = true;
        	        offload.enable = true;
                    # offload.offloadCmdMainProgram = "prime-run";

        	        nvidiaBusId = "PCI:1:0:0";
        	        intelBusId = "PCI:5:0:0";
                };
                hardware.nvidia.package = config.boot.kernelPackages.nvidiaPackages.stable;

                services.xserver.videoDrivers = [ "nvidia" ];

                environment.systemPackages = [ nvidia-offload ];
            };
        };
}