modules/nvidia/default.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

{ config, pkgs, lib, ... }:
{
  environment.systemPackages = with pkgs; [
    linuxPackages.nvidia_x11
    cudaPackages.cudnn
    cudaPackages.cudatoolkit
    cudaPackages.cuda_nvcc
  ];

  hardware = {
    graphics = {
      enable = lib.mkForce true;
    };

    nvidia = {
      open = true;
      prime = {
        sync.enable = true;
        intelBusId = "PCI:0:2:0";
        nvidiaBusId = "PCI:2:0:0";
      };
    };
  };

  nix = {
    settings = {
      substituters = [
        # "https://cache.nixos.org/" # set by default
        "https://cache.nixos-cuda.org"
      ];

      trusted-public-keys = [
        # "cache.nixos.org-1:6NCHdD59X431o0gWypbMrAURkbJ16ZPMQFGspcDShjY=" # set by default
        "cache.nixos-cuda.org:74DUi4Ye579gUqzH4ziL9IyiJBlDpMRn9MBN8oNan9M="
      ];
    };
  };

  nixpkgs.config = {
    allowUnfree = lib.mkForce true;
    cudaSupport = true;
    cudaForwardCompat = true;
  };

  services = {
    ollama = {
      acceleration = "cuda";
    };

    xserver = {
      enable = true;
      videoDrivers =[
        "nvidia"
      ];
    };
  };
}