diff --git a/modules/nvidia.nix b/modules/nvidia.nix new file mode 100644 index 0000000..6a6dd65 --- /dev/null +++ b/modules/nvidia.nix @@ -0,0 +1,20 @@ +{ config, pkgs, lib, ... }: + +{ + # 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; +}