30 lines
718 B
Nix
30 lines
718 B
Nix
{ pkgs, config, lib, ... }:
|
|
|
|
let
|
|
cfg = config.modules.system.nvidia;
|
|
in
|
|
{
|
|
options.modules.system.nvidia = {
|
|
enable = lib.mkEnableOption "enables nvidia";
|
|
};
|
|
|
|
config = lib.mkIf cfg.enable {
|
|
# enable nvidia proprietary driver
|
|
hardware.nvidia = {
|
|
modesetting.enable = true; # required
|
|
open = false; # use proprietary driver
|
|
nvidiaSettings = true; # no shit
|
|
powerManagement.enable = false; # can cause sleep issues
|
|
package = config.boot.kernelPackages.nvidiaPackages.stable;
|
|
};
|
|
|
|
# install nvidia-smi
|
|
environment.systemPackages = with pkgs; [
|
|
nvidia-smi
|
|
];
|
|
|
|
# enable docker gpu passthrough
|
|
virtualisation.docker.enableNvidia = true;
|
|
};
|
|
}
|