{ inputs, ... }:
{
  imports = [
    inputs.nixos-hardware.nixosModules.common-gpu-nvidia
    ./cuda-common.nix
  ];

  hardware.nvidia.prime = {
    # (required) Bus IDs. You can find them using lspci, grepping for "3D" or "VGA"
    # intelBusId  = "PCI:0:2:0";
    # nvidiaBusId = "PCI:1:0:0";

    offload.enable = true;
    offload.enableOffloadCmd = true;
    offload.offloadCmdMainProgram = "prime-run";
  };
}