{
  lib,
  config,
  pkgs,
  ...
}:
{
  config = lib.mkIf config.osbmModules.hardware.nvidia.enable {
    # Enable OpenGL
    hardware.graphics = {
      enable = true;
    };

    nixpkgs.config = {
      cudaSupport = true;
      cudaCapabilities = [ "8.9" ]; # 4090
    };

    # Load nvidia driver for Xorg and Wayland
    services.xserver.videoDrivers = [ "nvidia" ];

    hardware = {
      nvidia = {
        # Modesetting is required.
        modesetting.enable = true;

        # Nvidia power management. Experimental, and can cause sleep/suspend to fail.
        # Enable this if you have graphical corruption issues or application crashes after waking
        # up from sleep. This fixes it by saving the entire VRAM memory to /tmp/ instead
        # of just the bare essentials.
        powerManagement = {
          enable = false;

          # Fine-grained power management. Turns off GPU when not in use.
          # Experimental and only works on modern Nvidia GPUs (Turing or newer).
          finegrained = false;
        };
        # Use the NVidia open source kernel module (not to be confused with the
        # independent third-party "nouveau" open source driver).
        # Support is limited to the Turing and later architectures. Full list of
        # supported GPUs is at:
        # https://github.com/NVIDIA/open-gpu-kernel-modules#compatible-gpus
        # Only available from driver 515.43.04+
        # Currently alpha-quality/buggy, so false is currently the recommended setting.
        open = false;

        # Enable the Nvidia settings menu,
        # accessible via `nvidia-settings`
        nvidiaSettings = true;

        # Optionally, you may need to select the appropriate driver version for your specific GPU.
        package = config.boot.kernelPackages.nvidiaPackages.latest;
      };
      nvidia-container-toolkit.enable = lib.mkIf config.osbmModules.virtualisation.docker.enable true;
    };

    environment.systemPackages = [
      pkgs.nvidia-container-toolkit
    ];

    # TODO explain why this is needed
    programs.nix-required-mounts = {
      enable = true;
      presets.nvidia-gpu.enable = true;
    };

  };
}