nix-dotfiles/nixosModules/services/graphics/nvidia/default.nix

{ config, pkgs, lib, ... }:

{
  options = {
    nvidia.enable = lib.mkEnableOption
      "Enables nvidia with proprietary drivers";
  };

  config = lib.mkIf config.nvidia.enable {
    # Enable OpenGL
    hardware.opengl = {
      enable = true;
      driSupport = true;
      driSupport32Bit = true;
      extraPackages = with pkgs; [
        rocm-opencl-icd
        rocm-opencl-runtime
      ];
    };

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

    hardware.nvidia = {
      modesetting.enable = true;
      package = config.boot.kernelPackages.nvidiaPackages.beta;
    };

    boot.kernelParams = [ "nvidia_drm.fbdev=1" ];

    environment.systemPackages = with pkgs; [
      nvidia-vaapi-driver
    ];
  };
}