# Need nvidia proprietary drivers to work
{ pkgs, nixpkgs, config, lib, ... }:
{
  config = lib.mkIf config.frogeye.cuda {
    nix.settings = {
      substituters = [
        "https://cuda-maintainers.cachix.org"
      ];
      trusted-public-keys = [
        "cuda-maintainers.cachix.org-1:0dq3bujKpuEPMCX6U4WylrUDZ9JyUG0VpVZa7CNfq5E="
      ];
    };
    nixpkgs.overlays = [
      (self: super:
        let
          pkgs_uf = import nixpkgs {
            inherit (super) system;
            config.allowUnfree = true;
          };
        in
        {
          ctranslate2 = (pkgs_uf.ctranslate2.override {
            withCUDA = true;
            withCuDNN = true;
            stdenv = pkgs.gcc12Stdenv; # 24.11 Remove, as nvcc and gcc versions align again
            # Takes more than 16 GiB RAM to compile with the above, but fine if not
          });
        })
    ];
  };
}