{ pkgs, config, ... }:
{
  environment = {
    shellAliases.nvidia-settings = "nvidia-settings --config='$XDG_CONFIG_HOME'/nvidia/settings";

    systemPackages = with pkgs; [
      nvidia-container-toolkit
      nvidia-docker
    ];

    variables = {
      LIBVA_DRIVER_NAME = "nvidia";
      __GLX_VENDOR_LIBRARY_NAME = "nvidia";
      GBM_BACKEND = "nvidia-drm";
      __VK_LAYER_NV_optimus = "NVIDIA_only";
      __NV_PRIME_RENDER_OFFLOAD = "1";
      NVD_BACKEND = "direct";
      CUDA_PATH = "${pkgs.cudatoolkit}";
    };
  };

  boot = {
    blacklistedKernelModules = [ "nouveau" ];

    kernelParams = [
      "nvidia-drm.fbdev=1"
      "nvidia-drm.modeset=1"
      "nvidia.NVreg_PreserveVideoMemoryAllocations=1"
    ];
  };

  hardware = {
    nvidia-container-toolkit.enable = true;

    graphics = {
      extraPackages = [ pkgs.nvidia-vaapi-driver ];
      extraPackages32 = [ pkgs.nvidia-vaapi-driver ];
    };

    nvidia = {
      modesetting.enable = true;
      open = false;
      nvidiaSettings = true;
      package = config.boot.kernelPackages.nvidiaPackages.production;
      forceFullCompositionPipeline = true;

      powerManagement = {
        enable = true;
        finegrained = false;
      };
    };
  };
}